2021-12-24【编程谜题】【4个数之和】

最新推荐文章于 2025-12-20 22:59:04 发布

原创最新推荐文章于 2025-12-20 22:59:04 发布 · 334 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #动态规划

《编程谜题》同时被 2 个专栏收录

6 篇文章

订阅专栏

LeetCode

6 篇文章

订阅专栏

这篇博客介绍了如何使用双指针和深度优先搜索（DFS）算法解决寻找数组中四个数之和等于目标值的问题。提供了两种不同的实现方式，一种是双指针法，通过排序数组后进行两层循环，利用双指针调整和的大小；另一种是DFS法，通过递归遍历所有可能的组合，并进行剪枝优化，避免无效的计算。样例展示了算法的正确性和有效性。

文章目录

4个数之和

4个数之和

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[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 = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

提示：

$1 < = n u m s . l e n g t h < = 200$
$10^9 <= nums[i] <= 10^9$
$10^9 <= target <= 10^9$

题解

双指针

在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;

vector<vector<int>> fourSum(vector<int> &nums, int target)
{
    vector<vector<int>> result;
    sort(nums.begin(), nums.end());
    for (int k = 0; k < nums.size(); k++)
    {
        // 去重
        if (k > 0 && nums[k] == nums[k - 1])
            continue;
        for (int i = k + 1; i < nums.size(); i++)
        {
            // 正确去重方法
            if (i > k + 1 && nums[i] == nums[i - 1])
                continue;
            int left = i + 1;
            int right = nums.size() - 1;
            while (right > left)
            {
                //if (nums[k] + nums[i] + nums[left] + nums[right] > target)
                if (nums[k] + nums[i] > target - (nums[left] + nums[right]))
                    right--;
                else if (nums[k] + nums[i] < target - (nums[left] + nums[right]))
                    left++;
                else
                {
                    result.push_back(vector<int>{nums[k], nums[i], nums[left], nums[right]});
                    // 去重逻辑应该放在找到一个四元组之后
                    while (right > left && nums[right] == nums[right - 1])
                        right--;
                    while (right > left && nums[left] == nums[left + 1])
                        left++;
                    // 找到答案时，双指针同时收缩
                    right--;
                    left++;
                }
            }
        }
    }
    return result;
}

int main()
{
    vector<int> numbers = {1, 0, -1, 0, -2, 2};
    int target = 0;
    vector<vector<int>> result = fourSum(numbers, target);
    for (int i = 0; i < result.size(); i++)
    {
        for (int j = 0; j < result[i].size(); j++)
        {
            if (j == 0)
                cout << "(";
            printf("%2d ", result[i][j]);
            if (j != result[i].size() - 1)
                cout << ",";
            else
                cout << ")";
        }

        printf("\n");
    }
    return 0;
}

DFS+剪枝

在这里插入图片描述


#include <bits/stdc++.h>
using namespace std;

vector<vector<int>> ans;
vector<int> myNums, subans;
int tar, numSize;
void DFS(int low, int sum)
{
    if (sum == tar && subans.size() == 4)
    {
        ans.emplace_back(subans);
        return;
    }
    for (int i = low; i < numSize; ++i)
    {
        if (numSize - i < long(4 - subans.size()))
        { //剪枝
            return;
        }
        if (i > low && myNums[i] == myNums[i - 1])
        { //去重
            continue;
        }
        if (i < numSize - 1 && sum + myNums[i] + long(3 - subans.size()) * myNums[i + 1] > tar)
        { //剪枝
            return;
        }
        if (i < numSize - 1 && sum + myNums[i] + long(3 - subans.size()) * myNums[numSize - 1] < tar)
        { //剪枝
            continue;
        }
        subans.emplace_back(myNums[i]);
        DFS(i + 1, myNums[i] + sum);
        subans.pop_back();
    }
    return;
}

vector<vector<int>> fourSum(vector<int> &nums, int target)
{
    sort(nums.begin(), nums.end());
    myNums = nums;
    tar = target;
    numSize = nums.size();
    if (numSize < 4)
    {
        return ans;
    }
    DFS(0, 0);
    return ans;
}

int main()
{
    vector<int> numbers = {1, 0, -1, 0, -2, 2};
    int target = 0;
    vector<vector<int>> result = fourSum(numbers, target);
    for (int i = 0; i < result.size(); i++)
    {
        for (int j = 0; j < result[i].size(); j++)
        {
            if (j == 0)
                cout << "(";
            printf("%2d ", result[i][j]);
            if (j != result[i].size() - 1)
                cout << ",";
            else
                cout << ")";
        }

        printf("\n");
    }
    return 0;
}