leetcode打卡day4

三数之和
给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[ [-1, 0, 1],[-1, -1, 2]]

原来解法（通过了311/313个测试用例，超出执行时间）：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        Set<List<Integer>> result = new LinkedHashSet<>();
        int i,j,k;
        for(i=0;i<nums.length-2;i++){
            if(nums[i]>0)  break;
            for(j=i+1;j<nums.length-1;j++){
                for(k=j+1;k<nums.length;k++){
                    if(nums[i]+nums[j]+nums[k]==0){
                        List<Integer> value = Arrays.asList(nums[i], nums[j], nums[k]);
                    result.add(value);
                    }
                }
            }
        }
        return new ArrayList<>(result);
    }
}

改进后的解法（参考题解）：

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

思路：先通过sort对数组从小到大排序，nums[i]如果是正数就终止，然后用双指针j和k，从两端向中间运行，如果三数之和小于0，左指针j向右移动一位；三数之和大于0，右指针k向左移动一位；等于0的话则同时移动一位。