LeetCode 15 三數之和 / threeSum

題目描述:


此題如果純暴力的話應該是會超時的(沒試過)考慮在暴力的基礎上優化一下,先對數組排序,然後枚舉第一個數 i,那麼剩下兩個數之和爲 j + k = 0 - i,然後設立兩個指針j和k指向頭和尾,對剩下的數組從兩頭到中間搜索答案,時間複雜度爲o(n方),代碼如下:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        if(nums.size() < 3) return res;
        sort(nums.begin(),nums.end());
        int j,k,target;
        for(int i = 0; i < nums.size() - 2; i++){
            if(i != 0){
                if(nums[i] == nums[i-1]) continue;//避免重複搜索,比如 -2,-2,0,1,2。。。。。搜完第一個-2應跳過第二個-2
            }
            j = i + 1;
            k = nums.size() - 1;
            target = 0 - nums[i];
            while(j < k){
                if(nums[j] + nums[k] == target){
                    res.push_back({nums[i],nums[j],nums[k]});
                    while(nums[j] == nums[j+1] && nums[k] == nums[k-1]){//這裏是爲了避免形如 -2,0,0,2,2這種情況
                        j++;
                        k--;
                    }
                    j++;
                    k--;
                }
                else if(nums[j] + nums[k] < target) j++;
                else k--;
            }
        }
        return vector<vector<int>>(res.begin(),res.end());
    }
};


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章