LeetCode 15.三数之和

目录

题目描述

方法一

思路：

代码：

---

题目描述

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

提示：

• 3 <= nums.length <= 3000
• -105 <= nums[i] <= 105

方法一

思路：

双指针的题。先对数组进行排序，这样数组就有了单调性。最质朴的思路就是三重循环，但是时间复杂度太高 O(N3)，所以第三重循环变成一个从数组最右端开始向左移动的指针。（双指针我的硬骨头我啃啃啃）

首先，左指针指向第一个数的下一位，右指针从最后一位往前找。因为排序过后具有单调性，若当前的三个数和小于0，那右指针也不要继续往前了，只会更小。指针不动了有这些情况：

1.左右指针相遇了

2.找到三树之和为0了

3.三数之和小于0了

因此最后判断一下，是情况2就加入ans答案中。

有一个点需要注意，因为要求不重复的三元组，每个指针移动最初要判断一下是否和上一个元素相同，相同的话就往移动，直至不同。

代码：

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        int n=nums.length;
        Arrays.sort(nums);
        List<List<Integer>> ans=new ArrayList<List<Integer>>();
        // 枚举 a
        for(int first=0;first<n;first++){
            // 需要和上一次枚举的数不相同
            if(first>0&&nums[first]==nums[first-1]) continue;
            // c 对应的指针初始指向数组的最右端            
            int thrid=n-1;
            int target=0-nums[first];
             // 枚举 b
            for(int second=first+1;second<n;second++){
                 // 需要和上一次枚举的数不相同
                if(second>first+1&&nums[second]==nums[second-1]) continue;
                // 需要保证 b 的指针在 c 的指针的左侧
                while(second<thrid&&nums[second]+nums[thrid]>target){
                    thrid--;
                }
                // 如果指针重合，随着 b 后续的增加
                // 就不会有满足 a+b+c=0 并且 b<c 的 c 了，可以退出循环
                if(second==thrid) break;
                if(nums[second]+nums[thrid]==target) {
                    List<Integer> list=new ArrayList<Integer>();
                    list.add(nums[first]);
                    list.add(nums[second]);
                    list.add(nums[thrid]);
                    ans.add(list);
                }
            }
        }
        return ans;
    }
}