[Leetcode] 15 - 3Sum

LeetCode 3Sum 去重技巧

最新推荐文章于 2022-02-04 23:38:50 发布

原创最新推荐文章于 2022-02-04 23:38:50 发布 · 753 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

135 篇文章

订阅专栏

本文介绍了一种解决LeetCode经典题目3Sum的方法，并特别关注于如何有效地去除重复解。通过排序和双指针技巧，文章提供了一个高效的C++实现方案。

原题链接：https://oj.leetcode.com/problems/3sum/

经典3sum题，不过要注意这里的去重，去重是给原来这道题的基本形式加的难点。

1. 数组如果大小小于3，那就没有任何结果。

2. 将数组排序

3. 然后循环数组，每次都选取i所在的当前值为第一个结果，然后选取left = i + 1和right = size - 1。则问题变为num[left] + num[right]是否等于-num[i]。

4. 每次选取第一个元素的时候，要将其与之前的第一个元素相比，如果相同，则说明如果继续计算，此轮的计算结果已经被包含在上轮的计算结果之内，所以跳至下一轮。

5. 当left和right满足条件时候，我们需要将left和right移至下一个位置，这里需要将left之后和right之前的重复元素全部跳过，否则将会出现重复答案，则不符合题意。

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int> > res;
        
        if (num.size() < 3) {
            return res;
        }
        sort(num.begin(), num.end());
        int prev = INT_MIN;
        for (int i = 0; i < num.size() - 2; ++i) {
            if (num[i] == prev) {
                prev = num[i];
                continue;
            } else {
                prev = num[i];
            }
            
            int target = -num[i];
            int left = i + 1;
            int right = num.size() - 1;
            while (left < right) {
                int curSum = num[left] + num[right];
                if (curSum == target) {
                    vector<int> curRes;
                    curRes.push_back(num[i]);
                    curRes.push_back(num[left]);
                    curRes.push_back(num[right]);
                    res.push_back(curRes);
                    while (num[left] == num[left + 1] && left + 1 < num.size()) ++left;
                    while (num[right] == num[right - 1] && right - 1 >= 0) --right;
                    ++left;
                    --right;
                } else if (curSum < target) {
                    ++left;
                } else {
                    --right;
                }
            }
        }
        
        return res;
    }
};