三数之和有大学问

最新推荐文章于 2022-11-26 19:59:01 发布

踩着七彩祥云的猴子

最新推荐文章于 2022-11-26 19:59:01 发布

阅读量115

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/weixin_36149892/article/details/108934066

版权

leetcode 专栏收录该内容

148 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        if (n < 3) {
            return {};
        }
        unordered_set<int> us;
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size() - 2; i++) {
            while (i > 0 && i < nums.size() && nums[i] == nums[i-1]) {
                i++;
            }
            int l = i + 1;
            int r = nums.size() - 1;
            while (l < r) {
                int result = nums[l] + nums[r] + nums[i];
                if (result == 0) {
                    ans.push_back({nums[l], nums[r], nums[i]});
                    while (l < r && nums[r] == nums[r-1]) {
                        r--;
                    }
                    r--;
                    while (l < r && nums[l] == nums[l+1]) {
                        l++;
                    }
                    l++;
                }
                else if (result > 0) {
                    r--;
                }
                else {
                    l ++;
                }
            }
        }
        return ans;
    }
};