15. 三数之和

题目

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

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

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

思路

题干 ：三数之和为0，且不重复

• 双指针思路: 先对数组进行排序，然后固定三指针的最小值k，双指针i,j分别遍历(k,nums.length)首尾交替移动记录满足(nums[i]+nums[j]+nums[k])==0的组合
• 当nums[k]>0，直接跳出，因为nums[k]是最小值，那么肯定不存在满足i,j使得三数之和为0的序列
• 当k>0且nums[k]==nums[k-1]时后移跳过重复元素
• 当i<j且sum=nums[k]+nums[i]+nums[j]

1. s<0时，i++,并跳过重复元素
2. s>0时，j–并跳过所有的重复元素
3. s==0时，记录组合[k,i,j]到结果集res,执行i++,j–并跳过所有的重复元素

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        //排序
        Arrays.sort(nums);
        //结果集
        List<List<Integer>> res=new ArrayList<>();
        for(int k=0;k<nums.length-2;k++){
            //已升序排序，首元素都大于就不存在和为0的序列
            //这一步也可以没有
            if(nums[k]>0){
                break;
            }
            //跳过重复元素
            //这一步必须有，否则就会重复元素就会多计算
            if(k>0&&nums[k]==nums[k-1]){
                continue;
            }

            //定一动二:双指针
            int i=k+1;
            int j=nums.length-1;
            while(i<j){
                int sum=nums[i]+nums[j]+nums[k];
                if(sum>0){
                    //先减减再判重跳够重复元素
                    //while(i<j&&nums[j]==nums[--j]);可省略因为不相等i++就会跳过重复
                }else if(sum<0){
                    //先加加再判重跳过重复元素
                    while(i<j&&nums[i]==nums[++i]);
                }else{
                    //相等加入结果集
                    res.add(new ArrayList<Integer>(Arrays.asList(nums[i],nums[j],nums[k])));
                    //跳过重复元素
                    //没循环体直接分号结束表示空语句
                    while(i<j&&nums[i]==nums[++i]);
                    while(i<j&&nums[j]==nums[--j]);
                }
            }
        }
        return res;
    }
}

时间复杂度O(n^2) :k的时间复杂度O(n)，双指针i,j的时间复杂度O(n),总时间复杂度O(n^2)
空间复杂度O(log n) : 结果集res的空间复杂度不计入，但是排序的空间复杂度O(log n)

扩展：