18.四位数之和

给定一个包含 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]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题解:
排序+双指针

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> res;
        int n = nums.size();
        if (n < 4)
            return res;
        sort(nums.begin(), nums.end());

        int first, second, third, forth;
        for (first = 0; first < n - 3; first ++) {
            // 去除重复的a
            if (first > 0 && nums[first] == nums[first - 1])
                continue;
            if (nums[first] + nums[first + 1] + nums[first + 2] + nums[first + 3] > target)
                break;
            if (nums[first] + nums[n - 3] + nums[n - 2] + nums[n - 1] < target)
                continue;

            for (second = first + 1; second < n - 2; second ++) {
                // 去除重复的b
                if (second > first + 1 && nums[second] == nums[second - 1])
                    continue;
                if (nums[first] + nums[second] + nums[second + 1] + nums[second + 2] > target)
                break;
                if (nums[first] + nums[second] + nums[n - 2] + nums[n - 1] < target)
                    continue;
                third = second + 1;
                forth = n - 1;
                
                while (third < forth) {
                    int temp = nums[first] + nums[second] + nums[third] + nums[forth];
                    if (temp < target) {
                        third ++;
                    } else if (temp > target){
                        forth --;
                    } else {
                        res.push_back({nums[first], nums[second], nums[third], nums[forth]});
                        // 去重
                        while (third < forth && nums[third] == nums[third + 1])
                            third ++;
                        // 去重
                        while (forth > third && nums[forth] == nums[forth - 1])
                            forth --;
                        third ++;
                        forth --;
                    }
                }
            }
        }

        return res;
    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值