[LeetCode] 3Sum

最新推荐文章于 2022-01-27 22:44:32 发布

lyangpku

最新推荐文章于 2022-01-27 22:44:32 发布

阅读量411

点赞数

分类专栏： LeetCode 面试题算法 Array 文章标签： Array array Leetcode leetcode LeetCode 算法面试题

本文链接：https://blog.csdn.net/lyangpku/article/details/8553678

版权

LeetCode 同时被 3 个专栏收录

110 篇文章 0 订阅

订阅专栏

面试题

110 篇文章 0 订阅

订阅专栏

算法

110 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        vector<vector<int> > ret;
        if (num.empty()) return ret;
        
        sort(num.begin(), num.end());
        
        for (int i = 0; i < num.size(); i++) {
            if (i && num[i] == num[i - 1])
                continue;
                
            int a = num[i];
            
            int j = i + 1;
            int k = num.size() - 1;
            
            while (j < k) {
                int b = num[j];
                int c = num[k];
                
                if (a + b + c == 0) {
                    vector<int> sln;
                    sln.push_back(a);
                    sln.push_back(b);
                    sln.push_back(c);
                    ret.push_back(sln);
                    
                    for (j++; j < k && num[j] == num[j - 1]; j++);
                    for (k--; j < k && num[k] == num[k + 1]; k--);
                } else if (a + b + c < 0) {
                    for (j++; j < k && num[j] == num[j - 1]; j++);
                } else {
                    for (k--; j < k && num[k] == num[k + 1]; k--);
                }
            }
        }
        
        return ret;
    }
};

Small Case: 8ms

Large Case: 288ms

Time: O(n^2)

Space: O(n)