「リートコード - 1. Two Sum」

ジャバソリューション

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
class Solution {
public int[] twoSum(int[] nums, int target) {
class Foo {
int value;
int index;
Foo(int value, int index) {
this.value = value;
this.index = index;
}
};
int n = nums.length;
List<Foo> a = new ArrayList<Foo>();
for (int i = 0; i < n; ++i) {
a.add(new Foo(nums[i], i));
}
Collections.sort(a, new Comparator<Foo>() {
@Override
public int compare(Foo u, Foo v) {
return u.value - v.value;
}
});
int u = 0, v = n - 1;
while (u < v) {
if (a.get(u).value + a.get(v).value == target) {
return new int[]{a.get(u).index, a.get(v).index};
} else if (a.get(u).value + a.get(v).value < target) {
++u;
} else {
--v;
}
}
return new int[2];
}
}

总算是写了第一个,总的来说大概就是,Java还需要慢慢熟练,其实思路早就有了,但是看了题解发现比起排序的NlogN,用哈希N就可以,还是略败一筹。不过还是坚持写完,毕竟第一次用Java,而且主要是把Collections.sort这种东西写出来(Pair是真的难用,学学C++吧)。