每日打卡：三数之和

心情

今天又在简书上被安利了一部剧《大明王朝1566》
这才是可怕的历史，陈芊芊遇到这部剧，真活不过3集。

读题

leetcode： 15. 三数之和

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

测试用例：
输入:[-1, 0, 1, 2, -1, -4]
输出:[ [-1, 0, 1], [-1, -1, 2]]

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

乍一想，三层for循环？仔细一想，三层for循环……

思路

前提：数组先排过序，这样可以控制循环的次数
1：第一层循环，将第一个数固定
2：第二层循环，第二个数比第一个数大
3：第三层循环，找到唯一的解能满足三数之和为0
小技巧：第一个数固定的时候，第二个和第三个可以分别从两端向中间靠拢，直到两个数相遇。这样就算是3层循环，也是O(n²)的时间复杂度

实现

public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        //对数组升序排序
        Arrays.sort(nums);
        //用于标记最后一个数，j++的时候，从index处k--比较迅速
        int index,temp;
        for (int i = 0; i < nums.length - 2 && nums[i] <= 0; i++) {
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            index = nums.length - 1;
            //j从i+1到index，若nums[i] + nums[j] + nums[index] < 0则让j++
            for (int j = i + 1; j < index && (temp = nums[i] + nums[j]) <= 0; j++) {
                if (j > i + 1 && nums[j] == nums[j - 1] || temp + nums[index] < 0) {
                    continue;
                }
                //k从index到j，当nums[i] + nums[j] + nums[k] >= 0则让k--
                for (int k = index; k > j && (temp = nums[i] + nums[j] + nums[k]) >= 0; k--) {
                    if (temp == 0) {
                        result.add(Arrays.asList(nums[i], nums[j], nums[k]));
                        //标记当i固定时，k的最大下标，等待j再一次++
                        index = k;
                        break;
                    }
                }
            }
        }
        return result;
    }

为了让代码尽可能地行数少，写的比较冗长。只要注释足够就不影响理解。

提交

这已经是我优化了很多次的结果了。去掉了Set集合（用了相邻数不重复），去掉了存放结果的list（用了Arrays.asList()）。执行效率也提升了很多，就是名次没有提升……
第一次通过的代码如下：

public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> result = new ArrayList<>();
        List<Integer> element;
        Set<Integer> trimSet;
        //对数组升序排序
        Arrays.sort(nums);
        //先找出所有的解
        int temp = 0,index;
        //锚定第一个数
        for (int i = 0; i < nums.length - 2; i++) {
            if(nums[i] >0){
                break;
            }else if(i>0 && nums[i]==nums[i-1]){
                continue;
            }
            trimSet = new HashSet<>();
            index = nums.length - 1;
            //第二个数从i+1开始
            for (int j = i + 1; j < nums.length - 1; j++) {
                temp = nums[i] + nums[j];
                if(temp > 0){
                    break;
                }else if(!trimSet.add(nums[j])){
                    continue;
                }
                element = new ArrayList<>();
                element.add(nums[i]);
                element.add(nums[j]);
                //第三个数从后向前遍历
                for (int k = index; k > j; k--) {
                    temp = nums[i] + nums[j] + nums[k];
                    if (temp == 0) {
                        element.add(nums[k]);
                        result.add(element);
                        index = k;
                        break;
                    }else if(temp < 0){
                        break;
                    }
                }
            }
        }
        return result;
    }

标答

public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> lists = new ArrayList<>();
        //排序
        Arrays.sort(nums);
        //双指针
        int len = nums.length;
        for(int i = 0;i < len;++i) {
            if(nums[i] > 0) return lists;
            if(i > 0 && nums[i] == nums[i-1]) continue;
            int curr = nums[i];
            int L = i+1, R = len-1;
            while (L < R) {
                int tmp = curr + nums[L] + nums[R];
                if(tmp == 0) {
                    List<Integer> list = new ArrayList<>();
                    list.add(curr);
                    list.add(nums[L]);
                    list.add(nums[R]);
                    lists.add(list);
                    //两个指针向中间靠拢
                    while(L < R && nums[L+1] == nums[L]) ++L;
                    while (L < R && nums[R-1] == nums[R]) --R;
                    ++L;
                    --R;
                } else if(tmp < 0) {
                    ++L;
                } else {
                    --R;
                }
            }
        }
        return lists;
    }

标答取自评论区某大神
感觉差不多的思路啊，咋就差了呢？