首页 > 代码库 > [Leetcode] 15 - 3Sum

[Leetcode] 15 - 3Sum

原题链接: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;
    }
};


[Leetcode] 15 - 3Sum