15. 3Sum-数组

1题目理解

输入：int数组nums
规则：查找数组中任意三个数的和等于0
输出：符合条件的解，不包含重复的解
例子：
nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

2 排序+双指针

题解参考网址
做排序是因为解不能重复，而数组中元素可能是重复的。排序之后[-4,-1,-1,0,1,2]，在查找了组合(-4,-1,-1)、(-4,-1,0)…以后，就不能再查找(-4,-1,0)，这里的-1是下标为2的-1。否则就会有重复的解。

关键点2是双指针。排序好后的数组，当a确定以后b+c的值就是一个固定的值。b从小到大赋值，c从大到小赋值。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        int n = nums.length;
        Arrays.sort(nums);
        for(int i=0;i<n;i++){
            if(i==0 || nums[i] !=nums[i-1]){
                int a = nums[i];
                int k = n-1;//这里是重点
                for(int j = i+1; j<n;j++){
                    if(j==i+1 || nums[j] != nums[j-1]){
                        int b = nums[j];
                        while(k>j && nums[k]+a+b>0){
                            k--;
                        }
                        if(k!=j && nums[k]+a+b==0){
                            result.add(Arrays.asList(a,b,nums[k]));
                        }
                    }
                }
            }
            
            
        }
        return result;
    }
}

另外一种写法

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        int n = nums.length;
        Arrays.sort(nums);
        int i = 0;
        while(i<n){
            if(i>0  && nums[i-1]==nums[i]) {
                i++;
                continue;
            }
            int j = i+1,k=n-1;
            int target = 0 - nums[i];
            while(j<k){
                if(nums[j]+nums[k]==target){
                    result.add(Arrays.asList(nums[i],nums[j],nums[k]));
                    j++;
                    while(j<k && nums[j-1]==nums[j]) j++;
                    k--;
                }else if(nums[j]+nums[k]<target){
                    j++;
                }else{
                    k--;
                }
            }
            i++;
        }
        return result;
    }
}