LeetCode 热题 HOT 100-三数之和

最新推荐文章于 2024-06-03 22:30:56 发布

搬砖的肖傲楠

最新推荐文章于 2024-06-03 22:30:56 发布

阅读量323

点赞数

分类专栏： leetCode算法类文章标签：算法 leetcode 数据结构 java 动态规划

本文链接：https://blog.csdn.net/qq_43591363/article/details/112990079

版权

leetCode算法类专栏收录该内容

3 篇文章 0 订阅

订阅专栏

文章目录

LeetCode 热题 HOT 100题

LeetCode 热题 HOT 100题

前言

这个题对于我这种愚笨之人，看似简单但是题的难度确实为中等，思考了好长时间才做出来，啦啦啦！

三数之和题目

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例 1：
输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
示例 2：
输入：nums = []
输出：[]
示例 3：
输入：nums = [0]
输出：[]
提示：
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

JAVA实现代码

执行用时: 22 ms
内存消耗: 42.6 MB
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        int len = nums.length;
        List<List<Integer>> ans = new ArrayList<>();
        if (len < 3) return ans;
        for (int i = 0; i < len - 2; i++) {
            if (nums[i] > 0) break;
            if(i > 0 && nums[i] == nums[i-1]) continue;
            twoSum(nums, -nums[i], i + 1, len - 1, ans);
        }
        return ans;
    }
    void twoSum(int[] nums, int target, int start, int end, List<List<Integer>> ans) {
        while (start < end) {
            if (nums[start] + nums[end] == target) {
                List<Integer> res = new ArrayList<>();
                res.add(-target);
                res.add(nums[start]);
                res.add(nums[end]);
                ans.add(res);
                while(start < end && nums[start+1] == nums[start]) ++start;
                while(start < end && nums[end-1] == nums[end]) --end;
                ++start;
                --end;
            } else if (nums[start] + nums[end] < target) {
                ++start;
            } else {
                --end;
            }
        }
    }
}

题目分析

题目分析
首先分析题目：请你找出所有和为 0 且不重复的三元组。注意：答案中不可以包含重复的三元组。
- 示例 1:
  输入：nums = [-1,0,1,2,-1,-4]
  输出：[[-1,-1,2],[-1,0,1]]
过程分析

这个题我开始看的话感觉意思和不重复最长字符串字串的那个题目算法很类似，都是找最长或者最大集合返回出来，但是我最后仔细看题目要求还不一样，首先我们先看题目要求：找出所有和为0且不重复的三元组。这句话证明数组长度最少为3个所以程序开头就先判断了长度是否大于等于3：
```
int len = nums.length;
List<List<Integer>> ans = new ArrayList<>();
if (len < 3) return ans;  //首先判断数组长度如果小于3个则返回空集合
```
首先数组排一下序，通过Arrays.sort()完成，是数组变成有序数组，这个后面有用到，配合类似快速查找算法完成，当然快速查找算法的第一个条件就是数组或者集合有序，执行的循环次数是==(数组长度 - 2)== 。如果排完序num[i]>0,直接退出，因为后面的元素都是在怎么相加都不等于0了，肯定大于0；比如排完序为：[-2，1，1，2，3]，如果 i 为 1 的话，num[i] = 1的话，后面查找3个数在怎么比都不为零，还有一个条件就是不重复的三元组所以下面也判断了如果后一个元素和前一个元素相等的话，就算查出来为空的三个集合也和前一个集合相等，所以直接continue继续下一次循环执行，最后执行twoSum方法：两个数相加，num[i]相当于目标值，这里就用到了类似快速查找算法了。
```
Arrays.sort(nums);
int len = nums.length;
for (int i = 0; i < len - 2; i++) {
     if (nums[i] > 0) break;    
     if(i > 0 && nums[i] == nums[i-1]) continue;
     twoSum(nums, -nums[i], i + 1, len - 1, ans);
}
```
这个方法类似就是快速查询算法了，定义一个start和end，这个地方我说一下为什么是start = i+1, end = len-1,因为把num[i]当作目标值，比如：测试用例：nums = [-1,0,1,2,-1,-4]，i = 0，nums[i]就是 -1目标值，start = 1，end = 5; 从第2个数组下标 1 和数组尾 5 起取两个数做运算，判断是否两个数相加然后在加目标值是否为零，为零则加入到集合，然后在依次判断头节点和尾节点由外向内靠近。这一块的思想相当于：0为目标值，他会依次相加判断：0，1，2、0，1，3、0，1，4、0，1，5、0，2，3、0，2，4…0，4，5。这里加的判断头下标start和尾下标end变化就相当于简化了不相加这么多次了排除不符合要求的，节约时间和内存开销。
```
 void twoSum(int[] nums, int target, int start, int end, List<List<Integer>> ans) {
       while (start < end) {
           if (nums[start] + nums[end] == target) {
               List<Integer> res = new ArrayList<>();
               res.add(-target);
               res.add(nums[start]);
               res.add(nums[end]);
               ans.add(res);
               while(start < end && nums[start+1] == nums[start]) ++start;
               while(start < end && nums[end-1] == nums[end]) --end;
               ++start;
               --end;
           } else if (nums[start] + nums[end] < target) {
               ++start;
           } else {
               --end;
           }
       }
   }
```