Leetcode(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]
]

解体思路
呃呃呃额额,跟三数之和,两数之和类似的吧,可以使用双指针的方法。

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

猜你喜欢

转载自blog.csdn.net/weixin_43624053/article/details/84782321