其他分享
首页 > 其他分享> > 18. 四数之和

18. 四数之和

作者:互联网

给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:答案中不可以包含重复的四元组。

示例 1:

输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2:

输入:nums = [], target = 0
输出:[]

提示:

0 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

思路: 与三数之和完全相同。但需要注意的就是进行判断 确保相同的的结果不能重复出现。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        
        if(nums.size() < 4) return res;
        int num_len = nums.size();
        int a = 0, b = 1;
        for(;a < num_len - 3;a++)
        {
            if(a>0&&nums[a]==nums[a-1]) continue; //保证a不相同
            for(b=a+1;b<num_len - 2;b++)
            {
                if(b>a+1&&nums[b]==nums[b-1]) continue; //保证a不相同
                int c = b+1, d = num_len -1 ;
                while(c<d) //内层双指针
                {
                    int count = nums[a] +  nums[b] + nums[c] + nums[d] ;
                    if(count == target)
                    {
                        vector<int> temp;
                        temp.push_back(nums[a]);
                        temp.push_back(nums[b]);
                        temp.push_back(nums[c]);
                        temp.push_back(nums[d]);
                        res.push_back(temp);

                        while(c<d && nums[c]==nums[c+1]) c++;  // 这两步 保证 c d 移动不能重合
                        while(c<d && nums[d]==nums[d-1]) d--;
                        c++;
                        d--;
                    }
                    else if(count < target)
                    {
                        c++;
                    }
                    else
                    {
                        d--;
                    
                    }
                }
            }
        }
        return res;

   }
};

标签:四数,target,temp,nums,int,18,back,push
来源: https://blog.csdn.net/weixin_42493873/article/details/117855386