[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