Leetcode热题 15.三数之和

最新推荐文章于 2024-10-03 21:46:55 发布

jialun0116

最新推荐文章于 2024-10-03 21:46:55 发布

阅读量82

点赞数

分类专栏： leetcode 文章标签： leetcode 指针

本文链接：https://blog.csdn.net/qq_30031221/article/details/115691154

版权

leetcode 专栏收录该内容

208 篇文章 0 订阅

订阅专栏

本文介绍了一种使用双指针解决数组中寻找三数之和为0的元素的方法。首先对数组进行排序，然后通过一个外层循环和两个内层指针，动态调整指针位置以找到所有不重复的三元组。在找到符合条件的三元组后，利用指针移动进行去重操作。代码示例展示了具体的实现细节。

摘要由CSDN通过智能技术生成

15. 三数之和

知识点：双指针
时间：2021年4月14日
题目链接

题目
给你一个包含 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

解题思路

这道题和两数之和不同用哈希表比较难写和去重这里采用双指针移动的方法
为了找到不重复的三元组 首先对数组进行排序
最外面是一个循环i，注意需要去重和去重条件
内部定义两个指针，外循环和两个指针指向数字之和为0 就放入数组并更新两个指针进行去重操作
若小于0 则左边指针向右移动大于0 则右边的左移

代码

#include "cheader.h"
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        sort(nums.begin(),nums.end());
        if(nums.size() == 0 || nums[0] > 0)
            return ans;
        for(int i = 0; i < nums.size(); i++){
            if(i > 0 && nums[i] == nums[i-1]) //去重
                continue;
            int j = i+1;
            int k = nums.size()-1;
            while(j < k){
                if(nums[i] + nums[j] + nums[k] == 0){
                    ans.push_back({nums[i],nums[j],nums[k]});
                    while(j < k && nums[j] == nums[j+1]) j++; //和这个结果一样的数不要
                    while(j < k && nums[k] == nums[k-1]) k--;
                    j++;
                    k--;
                }else if (nums[i] + nums[j] + nums[k] < 0)
                    j++;
                else
                    k--;
            }
        }
        return ans;
    }
};