Leetcode 15. 三数之和

最新推荐文章于 2022-05-18 20:31:27 发布

Bendaai

最新推荐文章于 2022-05-18 20:31:27 发布

阅读量478

点赞数

本文链接：https://blog.csdn.net/Bendaai/article/details/79947556

版权

注意有相同的数出现的情况，需要跳过

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<int> tmp;
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());
        for (int i = 0; i + 2 < nums.size(); ++i) {
            if (i > 0 && nums[i] == nums[i - 1]) continue;//跳过
            int l = i + 1, r = nums.size() - 1, sum = -nums[i];//假定第一个元素
            while (l < r) {//找二元组，使第一个元素和二元组和为0
                while (l<r && nums[r] + nums[l] > sum) --r;
                if (l < r && nums[r] + nums[l] == sum) {//匹配的情况
                    tmp.clear();
                    tmp.push_back(nums[i]), tmp.push_back(nums[l]), tmp.push_back(nums[r]);
                    ans.push_back(tmp);
                }
                if (nums[l] == nums[r]) break;//跳过，注意要先加入答案集后再跳过
                while (l < r && nums[l] == nums[l + 1]) ++l;
                ++l;
            }
        }
        return ans;
    }
};