18. 四数之和

68 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

18. 四数之和

  • 题号:力扣18
  • 知识点:双指针
  • 总结 题干: 在这里插入图片描述

思路:

  • 1.相似题目三数之和
  • 2.最简单的想法就是四层循环,暴力解法,时间复杂度O(n4n^4)
  • 3.使用双指针将时间复杂度优化到O(n3n^3)
  • 4.两层循环来遍历得到ab,然后使用leftright来查找cd
  • 5.其他思路与三数之和相似。
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> ret;
        if(nums.size()<4){
            return ret;
        }
        for (int i = 0; i<nums.size(); i++){
            // if(nums[i]>target){
            //     return ret;
            // }
            // 结果去重
            if(i>0 && nums[i]==nums[i-1]){
                continue;
            }
            for (int j = i+1; j<nums.size(); j++){
                // if(nums[i]+nums[j]>target){
                //     return ret;
                // }
                // 结果去重
                if(j>i+1 && nums[j]==nums[j-1]){
                    continue;
                }
                int left = j+1;
                int right = nums.size()-1;
                while(right>left){
                	// nums[i]+nums[j]+nums[left]+nums[right]会溢出
                    if(nums[i]+nums[j]>target-nums[left]-nums[right]){
                        right--;
                        while(right>left && nums[right]==nums[right+1]) right--;
                    }
                    else if(nums[i]+nums[j]<target-nums[left]-nums[right]){
                        left++;
                        while(right>left && nums[left]==nums[left-1]) left++;
                    }
                    else{
                        vector<int> a = {nums[i],nums[j],nums[left],nums[right]};
                        ret.push_back(a);
                        right--;
                        left++;
                        while(right>left && nums[right]==nums[right+1]) right--;
                        while(right>left && nums[left]==nums[left-1]) left++;
                        
                        
                    }
                }
            }
        }
        return ret;
    }
};