Leetcode之4sum 解法

最新推荐文章于 2020-06-30 12:03:28 发布

原创最新推荐文章于 2020-06-30 12:03:28 发布 · 600 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种解决四数之和问题的高效算法实现。该算法基于3sum问题的解决方案，通过添加一层循环和双指针技巧来查找四个整数的组合，使得它们的和等于目标值。文章详细解释了如何避免重复解，并通过排序和跳过重复元素来优化算法性能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

类似3sum问题，在外面增加一个循环。
注意重复的问题。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int> &nums, int target) {
        vector<vector<int>> ans;
        if (nums.size() < 4)
            return ans;
        sort(nums.begin(), nums.end());
        int s1, s2, s3, s4;
        for (int i = 0; i < nums.size(); ++ i) {
            if (i != 0 && nums[i] == nums[i - 1])
                continue;
            for (int j = i + 1; j < nums.size(); ++ j) {
                if (j != i + 1 && nums[j] == nums[j - 1])
                    continue;
                findTarget(nums, ans, i, j, target - nums[i] - nums[j]);
            }
        }
        return ans;
    }
    void findTarget(const vector<int> &nums, vector<vector<int>> &ans,
                                   int i, int j, int target) {
        int left = i > j ? i + 1: j + 1, right = nums.size() - 1;
        while (left < right) {
            if (nums[left] + nums[right] < target)
                ++ left;
            else if (nums[left] + nums[right] > target)
                -- right;
            else {
                vector<int> v;
                v.push_back(nums[i]);
                v.push_back(nums[j]);
                v.push_back(nums[left]);
                v.push_back(nums[right]);
                ans.push_back(v);
                ++ left;
                while (left < right && nums[left] == nums[left - 1])
                    ++ left;
            }
        }
    }
};