15. 三数之和（重要）

题目

截图来自官方

 

这道题对于重复数据的消除的思想需要掌握

代码

class Solution {
    // 本题很经典，需要牢牢掌握。
    // 注意理解官方思路
    // 本质上大框架是先将给的数组从小到大排序，然后使用三重for循环遍历，遍历过程中跨过重复的数据来去除重复的结果。
    // 最终版本的代码可以对第二三层循环做优化，用双指针的思想让他们两个从左右逼近（因为first不变且second增加时，还想满足sum=0，则third必须减小）将第三层改为while循环来减小third，时间复杂度由O(n^3)降为O(n^2)。
    // 但是记着第一层for循环时要将third重新置为最右端。
    public List<List<Integer>> threeSum(int[] nums) {
        int n=nums.length;
        // third不能定义到这里，定义到这里会使最终结果少很多值。应该定义到第一个for中。例子：给定[-4,-3,-2,-1,-1,0,0,1,2,3,4]
        // int third=n-1;
        List<List<Integer>> res=new ArrayList();
        // 直接修改了nums,从小到大
        Arrays.sort(nums);
        for(int first=0;first<n;first++){
            // 下面的if逻辑错了，很细节。按下面来说，first为0时直接不进入了都。
            // if(first!=0&&nums[first]!=nums[first-1]){


                // 修改1
                if(first!=0&&nums[first]==nums[first-1]){
                    continue;
                }
                
                // 修改2
                int third=n-1;
                for(int second=first+1;second<n;second++){
                    // 同理，这里也错了
                    // if(second!=first+1&&nums[second]!=nums[second-1]){
                        // 是允许second==first+1的；比如【0，0，0】就是一组结果
                        if(second!=first+1&&nums[second]==nums[second-1]){
                            continue;
                        }           

                        while(second<third&&nums[first]+nums[second]+nums[third]>0){
                            third--;

                        }
                        // 元素不可以重合
                        if(third==second){
                            break;
                        }
                        // 必须有这个if，因为有可能上一个while后三个加起来直接小于0了（没有等于0的值）
                        if(nums[first]+nums[second]+nums[third]==0){
                            List<Integer> temp=new ArrayList();
                            temp.add(nums[first]);
                            temp.add(nums[second]);
                            temp.add(nums[third]);
                            res.add(temp);
                        }
                    // }
                }
            // }
        }
        return res;
    }
}