各种3sum 4sum

来源：互联网发布：java 首字母小写编辑：程序博客网时间：2024/05/16 16:14

这两天做了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;    }

0 0