1. 程式人生 > >Leetcode15.3Sum三數之和

Leetcode15.3Sum三數之和

給定一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。

注意:答案中不可以包含重複的三元組。

例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4], 滿足要求的三元組集合為: [ [-1, 0, 1], [-1, -1, 2] ]

雙指標加去重

class Solution {
public:
    vector<vector<int> > threeSum(vector<int>& nums) {
        int len = nums.size();
        sort(nums.begin(), nums.end());
        map<int, pair<int, int> > check;
        vector<vector<int> > res;
        for(int i = 0; i < len - 2; i++)
        {
            int low = i + 1;
            int high = len - 1;
            while(low < high)
            {
                if(nums[low] + nums[high] == -nums[i])
                {
                    res.push_back({nums[i], nums[low], nums[high]});
                    //去重
                    while(nums[low] == nums[low + 1])
                        low++;
                    low++;
                }
                if(nums[low] + nums[high] > -nums[i])
                {
                    high--;
                }
                else
                {
                    low++;
                }
            }
            //去重
            while(nums[i] == nums[i + 1])
                i++;
        }
        return res;
    }
};