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());
    }
};


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