leetcode刷题--15. 三数之和

题目描述（题目链接）

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

注意：答案中不可以包含重复的三元组。

示例 ：

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

求解思路

啊这道题竟然两年前就做过，但还是没想出来

先对数组排序，从i=0开始，用双指针j和k遍历i后面的数

• 情况1：若nums[i]+nums[j]+nums[k]>0，k--（这样num[k]就变小了）
• 情况2：若nums[i]+nums[j]+nums[k]<0，j--（这样num[j]就变大了）
• 情况3：若nums[i]+nums[j]+nums[k]=0，得到一个解

注意：要去重

代码

执行用时：23 ms, 在所有 Java 提交中击败了80.50%的用户

内存消耗：42.1 MB, 在所有 Java 提交中击败了87.88%的用户

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        if(nums.length<3){
            return result;
        }
        Arrays.sort(nums);                      //排序
        for(int i=0;i<nums.length-2;i++){
            if(nums[i]>0){                       //后面的肯定也大于0，不用再遍历了
                break;
            }
            if(i>0&&nums[i]==nums[i-1]) continue;     //去重
            int t = -nums[i];
            int j = i+1;
            int k = nums.length-1;
            while(j<k){
                if(nums[j]+nums[k]>t){              //情况1
                    k--;
                }else if(nums[j]+nums[k]<t){        //情况2
                    j++;
                }
                else{                             //情况3
                    result.add(new ArrayList<>(Arrays.asList(nums[i],nums[j],nums[k])));
                    j++;
                    k--;
                    while(j<k&&nums[j]==nums[j-1]) j++;     //去重
                    while(j<k&&nums[k]==nums[k+1]) k--;     //去重
                }
            }
        }
        return result;
    }
}