【7.5】15. 三数之和

最新推荐文章于 2024-07-25 19:34:38 发布

howtoloveyou

最新推荐文章于 2024-07-25 19:34:38 发布

阅读量49

点赞数

分类专栏： Leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_38758049/article/details/125625025

版权

Leetcode 专栏收录该内容

23 篇文章 0 订阅

订阅专栏

对vector快排为从小到大：sort(nums.begin(), nums.end());
对vector快排为从大到小：sort(nums.begin(), nums.end(), greater<int>());

#define MAX_INF 1e7
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> ret;
        vector<int> ans;
        vector<int> pre_ans;
        int a = -MAX_INF, b, c;
        int b_ptr, c_ptr;
        if(nums.size() < 3) return ret;

        for(int i = 0; i < nums.size() - 2; i++) {
            if(a != nums[i]) {
            a = nums[i];
            b = nums[i + 1];
            b_ptr = i + 1;
            c = nums[nums.size() - 1];
            c_ptr = nums.size() - 1;
            while(b_ptr < c_ptr) {
                //printf("a: %d, b: %d, c: %d\n", a, b, c);
                if((a + b + c) < 0) {
                    b_ptr++;
                    b = nums[b_ptr];
                } else if((a + b + c) > 0) {
                    c_ptr--;
                    c = nums[c_ptr];
                } else {
                    ans.push_back(a);
                    ans.push_back(b);
                    ans.push_back(c);
                    if(pre_ans.size() == 0 || pre_ans != ans) {
                        pre_ans = ans;
                        ret.push_back(ans);
                        ans.pop_back();
                        ans.pop_back();
                        ans.pop_back();
                    } else if(pre_ans == ans) {
                        ans.pop_back();
                        ans.pop_back();
                        ans.pop_back();
                    }
                    b_ptr++;
                    b = nums[b_ptr];
                }
            }}
        }
        return ret;
    }
};