LeetCode 15 — 3Sum(三数之和)

最新推荐文章于 2022-09-03 15:08:51 发布

卓尔货币

最新推荐文章于 2022-09-03 15:08:51 发布

阅读量315

点赞数

CC 4.0 BY-SA版权

分类专栏：算法笔记文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/s1293678392/article/details/80466031

算法笔记专栏收录该内容

36 篇文章

订阅专栏

本文介绍了一种算法，用于在给定数组中查找所有唯一且不重复的三元组，这些三元组的和为零。通过排序数组并使用双指针技巧，有效地解决了这个问题，并提供了C++实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
Note:
The solution set must not contain duplicate triplets.

Example:
Given array nums = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

翻译
给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

分析
先将数组排序，再一个数从头遍历，剩下的两个数从首尾靠近，找三数之和为target的，值得注意的是去除重复。

c++实现

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if (nums.size() < 3)
            return res;
        vector<int> tmp;
        sort(nums.begin(),nums.end());
        for (int i = 0; i < nums.size()-2; 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)
                    j++;
                else if (nums[i]+nums[j]+nums[k] > 0)
                    k--;
                else
                {
                    tmp.clear();
                    tmp.push_back(nums[i]);
                    tmp.push_back(nums[j]);
                    tmp.push_back(nums[k]);
                    res.push_back(tmp);
                    j++;
                    k--;
                    while (j < k && nums[j] == nums[j-1])
                        j++;
                    while (j < k && nums[k] == nums[k+1])
                        k--;
                }
            }
        }
        return res;
    }
};