Leetcode 第15题 三数之和题解

Leetcode 第15题 三数之和题解

题目描述

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

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

来源：力扣（LeetCode）
链接：LeetCode

题解

（1）拿到此类题目肯定不要第一时间考虑暴力解，此题考虑暴力解，要至少三层循环，时间复杂度，一下干到O(n³)，而且也不能很好保证上述所说的不重复，这显然不是一个好的解；

（2）那么问题来了，什么是好的解呢；

（3）且慢慢分析来，乍一看，这道题跟两数之和有点像，当时解两数之和这个题，有用HashMap的，有暴力两重循环解的，还有用双指针的，眼前一亮，双指针解三数之和行不行？

（4）试试双指针，但是为了解题方便且保证解不重复，先排个序；

（5）然后采用同样的思路，从两边向中间逼近，什么意思呢，就是定位当前元素后，得到三数中一个数，那剩余两个数怎么确定呢，好说，肯定不能再看当前元素前面的元素了，因为这样可能会导致重复。那OK，看后面的元素，由于要从两边逼近，那最后一个元素肯定要考虑，因此定位最右边元素为最后一个元素，那左边向右逼近的当然就是当前元素的下一个元素啦！因为这样才不会漏掉元素嘛；

（6）至此，三个元素都确定了，那接下来的事就简单了，判断三数之和是否为0，如果是，那么OK，将三个数作为一个list添加到结果中，如果小于0，此时肯定是要让它大一点接近0，因此左边向右逼近（上述已说明是在排好序的前提下），同样地如果大于0，那就是太大了，让他变小一点，右向左逼近。

（7）这就得到了整体的思路，要遍历数组，当然就是循环啦，具体细节看代码

代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        int len = nums.length;
        Arrays.sort(nums);
        if(nums.length<3||nums[0]>0||nums[len-1]<0)//先判断特殊情况
            return res;
        for(int i=0;i<len;i++){
            if(nums[i]>0) return res;//由于是排好序的，当当前元素大于零时，就不必再判断了
            if(i>0&&nums[i]==nums[i-1]) continue;//过滤重复元素可能得到的重复解
            int curr=nums[i];//当前元素
            int left=i+1;//左边
            int right=len-1;//右边
         
            while(left<right){//向中间靠拢
                List<Integer> list = new ArrayList<>();
                int tmp=curr+nums[left]+nums[right];
                if(tmp==0){
                    list.add(curr);
                    list.add(nums[left]);
                    list.add(nums[right]);
                    res.add(list);
                    while(left<right&&nums[left]==nums[left+1]) left++;
                    while(left<right&&nums[right]==nums[right-1]) right--;
                    left++;
                    //当前退出，是因为nums[left]!=nums[left+1],但是nums[left]=nums[left-1]
                    //因此让left再向右一步，这样就是不同的元素啦,下面right同理
                    right--;
                }
                else if(tmp>0){
                    right--;
                }
                else{
                    left++;
                }
            }
        }
        return res;
    }
}