LeetCode练习24：数组中和为0的三个数

LeetCode链接：力扣

题目：

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a ，b ，c ，使得 a + b + c = 0 ？请找出所有和为 0 且 不重复 的三元组。

示例：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

代码：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> ans = 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]){
                continue;
            }
            int left = i + 1, right = n - 1;
            while(left < right){
                if(nums[i] + nums[left] + nums[right] > 0)  right--;
                else if(nums[i] + nums[left] + nums[right] < 0)  left++;
                else if(nums[i] + nums[left] + nums[right] == 0){
                    List<Integer> list = new ArrayList<Integer>();
                    list.add(nums[i]);
                    list.add(nums[left]);
                    list.add(nums[right]);
                    ans.add(list);
                    //去重
                    while(right > left && nums[right] == nums[right-1]) right--;
                    while(right > left && nums[left] == nums[left+1])   left++;
                    right--;
                    left++;
                }
                
            }
        }
        return ans;
    }
}