[Leetcode] 4Sum

描述

Given an array S of n integers, are there elements a, b, c, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.

Note: The solution set must not contain duplicate quadruplets.

For example, given array S = [1, 0, -1, 0, -2, 2], and target = 0.

A solution set is:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

给定一个数组以及一个目标值，找出所有四元组，它们的和为目标值。

分析

参考 Two Sum 以及 3Sum 这两个问题，其实后一个问题是在前一个问题的基础上加上一层循环，对应的这个问题也是在“3Sum”这个问题的基础上再加一层循环。

建议大家把相关的这几道题目放在一起研究。

代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        set<vector<int>> res;
        sort(nums.begin(), nums.end());
        for (int l = 0; l < nums.size(); l++) {
            for (int k = l + 1; k < nums.size(); k++) {
                int i = k + 1, j = nums.size() - 1, sum = target - nums[l] - nums[k];
                while (i < j) {
                    if (nums[i] + nums[j] == sum) 
                        res.insert({nums[l], nums[k], nums[i++], nums[j--]});
                    if (nums[i] + nums[j] < sum) i++;
                    else if (nums[i] + nums[j] > sum) j--; 
                }
            }
        }
         return vector<vector<int> > (res.begin(), res.end());
    }
};

相关问题

Two Sum
3Sum
3Sum Closest