lotus

贵有恒何必三更眠五更起 最无益只怕一日曝十日寒

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

1. 题目

 

https://leetcode.cn/problems/3sum/

 

 

 

2. 解法

思路是这样的:

  • 首先,对数组进行排序,这样可以方便地跳过重复的元素,也可以利用双指针法来寻找第二个数和第三个数。
  • 然后,遍历数组,固定第一个数,如果第一个数大于零,那么后面的数都大于零,不可能有解,所以可以直接结束循环。
  • 接着,使用双指针法,从左右两端向中间寻找第二个数和第三个数,如果三数之和为零,就找到一个解,并跳过重复的元素;如果三数之和小于零,就说明第二个数太小,需要增大;如果三数之和大于零,就说明第三个数太大,需要减小。
  • 最后,返回结果列表。

第一元素固定+后两个元素采用双指针

public List<List<Integer>> threeSum(int[] nums) {
    //先对数组进行排序
    Arrays.sort(nums);
    //创建一个结果列表
    List<List<Integer>> result = new ArrayList<>();
    //遍历数组,从第一个数开始,到倒数第三个数结束
    for (int i = 0; i < nums.length - 2; i++) {
        //如果当前数和前一个数相同,就跳过,避免重复
        if (i > 0 && nums[i] == nums[i - 1]) {
            continue;
        }
        //定义两个指针,分别指向当前数后面的第一个数和最后一个数
        int left = i + 1;
        int right = nums.length - 1;
        //当左指针小于右指针时,循环查找
        while (left < right) {
            //计算当前三个数的和
            int sum = nums[i] + nums[left] + nums[right];
            //如果和为0,就找到了一个解
            if (sum == 0) {
                //把这三个数加入到结果列表中
                result.add(Arrays.asList(nums[i], nums[left], nums[right]));
                //左指针右移,跳过相同的数,避免重复
                while (left < right && nums[left] == nums[left + 1]) {
                    left++;
                }
                //右指针左移,跳过相同的数,避免重复
                while (left < right && nums[right] == nums[right - 1]) {
                    right--;
                }
                //左指针和右指针都移动一位,继续查找
                left++;
                right--;
            } else if (sum > 0) {
                //如果和大于0,说明右边的数太大了,需要减小,所以右指针左移
                right--;
            } else {
                //如果和小于0,说明左边的数太小了,需要增大,所以左指针右移
                left++;
            }
        }
    }
    //返回结果列表
    return result;
}

 其中以下三处 处理重复 非常重要

        //如果当前数和前一个数相同,就跳过,避免重复
        if (i > 0 && nums[i] == nums[i - 1]) {
            continue;
        }
                //左指针右移,跳过相同的数,避免重复
                while (left < right && nums[left] == nums[left + 1]) {
                    left++;
                }
                //右指针左移,跳过相同的数,避免重复
                while (left < right && nums[right] == nums[right - 1]) {
                    right--;
                }

 

3. 总结

posted on 2023-04-18 16:05  白露~  阅读(9)  评论(0编辑  收藏  举报