四数之和

题意:给定一个有N个元素的整数数组nums和一个整数target。求数组中四个元素之和等于target的组合,要求组合不能重复。

 

思路:首先对数组进行排序,因为要保证组合不能重复。设立四个“指针”a,b,c,d。其中a,b固定在数组最开头,c = b+1,d位于数组末尾,之后通过移动指针c和d使得nums[a],nums[b],nums[c],nums[d]之和等于target。当指针c和d重合时表明,以nums[a],nums[b]为开头的组合已经寻找完毕。所以移动指针b,当指针b移至末尾再移动指针a。

注意:因为数组不保证数据唯一,所以在移动指针时要避免移动后的数据不发生改变。

代码:

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        int len = nums.size();
        if(len < 4) return ans;
        sort(nums.begin(), nums.end());
        for(int a=0;a<len - 3;a++)
        {
            while(a > 0 && nums[a] == nums[a - 1]) a++;
            for(int b=a+1;b<len - 2;b++)
            {
                while(b > a + 1 && nums[b] == nums[b - 1]) b++;
                int c = b+1, d = len - 1;
                
                if(c == len -1) continue;
                while(c < d)
                {
                    while(c < d && (nums[a] + nums[b] + nums[c] + nums[d]) < target) c++;
                    while(c < d && (nums[a] + nums[b] + nums[c] + nums[d]) > target) d--;
                    if(c < d && (nums[a] + nums[b] + nums[c] + nums[d]) == target)
                    {
                        ans.push_back({nums[a], nums[b], nums[c], nums[d]});
                        c++;
                        while(c < d && nums[c] == nums[c -1]) c++;
                    }
                }
            }
        }
        return ans;
    }
};
//-2 -1 0 0 1 2

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值