码迷,mamicode.com
首页 > 其他好文 > 详细

四数之和

时间:2021-06-02 14:03:10      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:重复   class   cto   color   solution   end   rgba   存在   tor   

给定一个包含 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
输出:[]

class Solution {
public:
    vector <vector<int>> fourSum(vector<int> &nums, int target) {
        sort(nums.begin(), nums.end());
        vector <vector<int>> res;
        int size = nums.size();
        if (size < 4)
            return res;
        else if (size == 4 && nums[0] + nums[1] + nums[2] + nums[3] == target) {
            res.push_back(nums);
            return res;
        }
        for (int i = 0; i < size - 3; ++i) {
            if (i > 0 && nums[i] == nums[i - 1])
                continue;
            if (nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target)
                break;
            if (nums[i] + nums[size - 3] + nums[size - 2] + nums[size - 1] < target)
                continue;
            for (int j = i + 1; j < size - 2; ++j) {
                if (j > i + 1 && nums[j] == nums[j - 1])
                    continue;
                if (nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target)
                    break;
                if (nums[i] + nums[j] + nums[size - 2] + nums[size - 1] < target)
                    continue;
                int l = j + 1;
                int r = size - 1;
                while (l < r) {
                    if (nums[l] + nums[r] == target - nums[i] - nums[j]) {
                        res.push_back({nums[i], nums[j], nums[l], nums[r]});
                        while (l < r && nums[l] == nums[l + 1])
                            l++;
                        l++;
                        while (l < r && nums[r] == nums[r - 1])
                            r--;
                        r--;
                    } else if (nums[l] + nums[r] > target - nums[i] - nums[j])
                        r--;
                    else
                        l++;
                }
            }
        }
        return res;
    }
};

 

四数之和

标签:重复   class   cto   color   solution   end   rgba   存在   tor   

原文地址:https://www.cnblogs.com/zhangzhangtabszj/p/14821011.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!