[LeetCode] 3Sum 三数之和[medium]

最新推荐文章于 2022-03-04 23:23:06 发布

原创最新推荐文章于 2022-03-04 23:23:06 发布 · 120 阅读

0 ·

CC 4.0 BY-SA版权

算法设计与分析课程专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种解决LeetCode上3Sum问题的有效方法，该问题要求在整数数组中找到所有唯一的三元组，使得这三个数相加等于零。文章详细解释了通过排序和双指针技术来优化解决方案，以降低时间复杂度并去除重复结果。

[LeetCode] 3Sum 三数之和

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note:

Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
The solution set must not contain duplicate triplets.

    For example, given array S = {-1 0 1 2 -1 -4},

    A solution set is:
    (-1, 0, 1)
    (-1, -1, 2)

这道题做了之后发现难点其实不在于寻在三个和为0的数而在于对结果去重。寻在三个和为0的问题可以1. 直接遍历用三重循环做，反正这次leetcode也没时间复杂度限制，也可以2.对结果排序后固定一个值a，求满足和为－a的另外两个值。这样时间复杂度为o（n2）。并对结果进行去重处理。代码如下：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        sort(nums.begin(), nums.end());
        for (int k = 0; k < nums.size(); ++k) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k] == nums[k - 1]) continue;
            int target = 0 - nums[k];
            int i = k + 1, j = nums.size() - 1;
            while (i < j) {
                if (nums[i] + nums[j] == target) {
                    res.push_back({nums[k], nums[i], nums[j]});
                    while (i < j && nums[i] == nums[i + 1]) ++i;
                    while (i < j && nums[j] == nums[j - 1]) --j;
                    ++i; --j;
                } else if (nums[i] + nums[j] < target) ++i;
                else --j;
            }
        }
        return res;
    }
};