这两天做了3sum, 3sum closest, 4sum
一个模板就是先sort array, 取一个或两个下标(depend on 3sum or 4sum)然后用另外两个指针指向头和尾。由于数组已经排序,只需要这两个指针扫一遍。
教训:当sum == target时,经常忘记移动两个下标(k,l),使得死循环或者不再去判断k+1到l-1之间的数组。
2015.1 update:
对于有重复数字的情况,方法就是在每次循环后将下标一直移动到不重复的地方。注意不可以在循环前判断,比如{0,0,0,0}, 0 这样的情况就会返回空,因为第二层循环发现自己与前一个数字一样就会跳过。
public List<List<Integer>> fourSum(int[] num, int target) {
Arrays.sort(num);
List<List<Integer>> result = new ArrayList<List<Integer>>();
for (int i = 0; i < num.length-3; i++) {
if (i > 0 && num[i] == num[i-1]) {
continue;
}
for (int j = i+1; j < num.length-2; j++) {
int twoSum = target - num[i] - num[j];
int p = j+1;
int q = num.length-1;
while(p < q) {
if (num[p] + num[q] == twoSum) {
List<Integer> list = new ArrayList<Integer>();
list.add(num[i]);
list.add(num[j]);
list.add(num[p]);
list.add(num[q]);
result.add(list);
p++;
q--;
while (p < q && num[p] == num[p-1]) {
p++;
}
while (p < q && num[q] == num[q+1]) {
q--;
}
} else if (num[p] + num[q] < twoSum) {
p++;
} else {
q--;
}
}
while (j < num.length-2 && num[j] == num[j+1]) {
j++;
}
}
}
return result;
}