Leetcode 18 - 4Sum

原创于 2017-02-23 11:48:10 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 同时被 2 个专栏收录

157 篇文章

订阅专栏

array

24 篇文章

订阅专栏

本文介绍了一种解决四数之和问题的高效算法。通过先对数组进行排序，然后枚举两个数，并将问题转化为两数之和，最终达到O(n³)的时间复杂度。文章提供了详细的代码实现。

题意

给一个数组和target，要求在数组中找到四个数，使a + b + c + d = target。

思路

类似于3Sum，排序后枚举其中两个数a和b，然后问题就转化成了2Sum。

时间复杂度 $O(n^3)$

代码

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