leetcode刷题之四数之和

最新推荐文章于 2024-09-14 22:25:23 发布

李奥去哪了

最新推荐文章于 2024-09-14 22:25:23 发布

阅读量109

点赞数

分类专栏：算法刷题文章标签：算法

本文链接：https://blog.csdn.net/qq_19094871/article/details/106313178

版权

算法刷题专栏收录该内容

14 篇文章 0 订阅

订阅专栏

四数之和

题目
解题思路
- 纯暴力
- 排序+双指针
解题代码
提交结果
反思

题目

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。
注意：
答案中不可以包含重复的四元组。
示例：

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

满足要求的四元组集合为：
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

解题思路

纯暴力

暴力求解，就是四个for循环，遍历数组求解所有的四元组的和是否等于target。这种方法时间复杂度贼高， $O(n^4)$ 。这肯定不是我想要的。所以需要在这种基础上进行优化！

排序+双指针

如果你做过三数之和、两数之和，就应该明白套路了，稳定一个数，找另一个目标数的时候，直接顺序搜索的话，时间复杂度为n，如果搞成hash表的话，时间复杂度为1。那么这题怎么搞！

其实很简单，前面一个for稳定第一个数，接着第二个for稳定第二个数，接下来找余下两个数的时候，怎样将时间复杂度降到n呢，就是遍历余下数一次就找出了所有的符合要求的两个数呢？

答案是先排序，用左指针left指向最左的元素，right指向最右的元素，如果前面两个数加上这两个指针值小于target，那么说明数要大一点，那就 $+ + l e f t$ ；如果总和大于 $t a r g e t$ ，那就 $- - r i g h t$ ，这样left如果与right重合，说明余下的数已经遍历完，而且组合都选出来了。接下来就顺序移动第二个数，然后再重置left,right，这样玩！这样时间复杂度降到了 $o(n^2)$

解题代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        if(nums.size() < 4){
            return ans;
        }
        sort(nums.begin(),nums.end()); // 排序，方便后面双指针
        int one(0);
        while(one < nums.size()-3){
            int two = one + 1;
            if(nums[two] >= 0 && nums[two]+nums[one] > target){
                    break;
            }

            while(two < nums.size()-2){
                int left = two+1;
                int right = nums.size()-1;
                if(nums[two] >=0 && nums[two]+nums[one] > target){
                    break;
                }
                while(left < right && right < nums.size()){
                    if(nums[one]+nums[two]+nums[left]+nums[right] < target){
                        ++left;
                        while(left < right && nums[left] == nums[left-1]){
                            ++left;
                        }
                    }else if (nums[one]+nums[two]+nums[left]+nums[right] == target){
                        vector<int> res{nums[one],nums[two],nums[left],nums[right]};
                        ans.push_back(res);
                        ++left;
                        --right;
                        while(left < right && nums[right] == nums[right+1] && nums[left] == nums[left-1]){
                            --right;
                            ++left;
                        }
                    }else{
                        --right;
                        while(left < right && nums[right] == nums[right+1]){
                            --right;
                        }
                    }
                }
                ++two;
                while(two < nums.size()-2 && nums[two] == nums[two-1]){
                    ++two;
                }
                
            }
            ++one;
            while(one < nums.size()-3 && nums[one] == nums[one-1]){
                ++one;
            }
        }
        return ans;
    }
};