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

問題描述

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

問題分析

基本思路就是先對數組排序,順序選擇一個數字,然後使用左右指針,依次對剩下數據進行遍歷,考慮到結果會有重複,所以,要注意如何去重,詳細思路請見源代碼。

源代碼

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) 
    {
        vector<vector<int>> res;
        if(nums.size() <= 2) return res;
        sort(nums.begin(), nums.end());
        int l,r;
        int sum=0;
        for(int i=0;i<nums.size()-2;i++)
        {
            if(nums[i]>0) return res;
            //如果前後兩個數相同的話,那麼最終找出的另外兩個數結果一定是相同的,所以可以直接跳過
            if(i>0&&nums[i]==nums[i-1]) continue;
            l=i+1;
            r=nums.size()-1;
            while(l<r)
            {
                sum=nums[i]+nums[l]+nums[r];
                if(sum==0)
                {
                    
                    res.push_back({nums[i],nums[l],nums[r]});
                    l++;
                    r--;
                    //這裏也是爲了去重,在以i爲基礎,搜索後面兩個數的時候,面對重複的可以直接跳過,因爲最終得到的結果一定是相同的
                    while(l<r&&nums[l]==nums[l-1]) l++;
                    while(l<r&&nums[r]==nums[r+1]) r--;
                }
                else if(sum<0)
                {
                    l++;
                }
                else
                {
                    r--;
                }
            }
        }
        return res;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章