[leetcode] 18. 4Sum

18. 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]
]

题意：
      给定一个含有n个整数的数组，在给定的数组中，找到四个数，使其之和等于目标值。在给定数组中找到所有的等于目标值的不相同的四元组。
      注意：答案要求四元组集合不包含重复的元素。

思路：
      这道题的思路和之前的 [leetcode] 15. 3Sum 差不多，首先对数组排序，确定四元数中的前两个（a，b）， 遍历剩余数组确定两外两个（c，d），确定cd时思路跟 [leetcode] 15. 3Sum 确定后两个数据一样，二分查找左右逼近，在去重时采用set集合。用两个指针pre和tail从b这个数后面的两边往中间查找，用第一重for循环确定第一个数字，用第二重for循环确定第二个数字，然后在while循环里，第三、第四个数字分别从两端往中间扫。当这四个数之和等于target的时候，记录一下，当之和大于target的时候，tail左移，而当之和小于target的时候，pre右移，直到pre和tail相遇。这个其实就是在3Sum外层加1层循环。 代码如下：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        set<vector<int>> res2;//去重
        vector<vector<int>> res;
        sort(nums.begin(), nums.end());
        int len = nums.size();
        for (int i = 0;i<len-3;i++)
        {
            for (int j = i+1;j<len-2;j++)
            {
               int pre = j + 1;
               int tail = len - 1;
               while (pre < tail)
               {
                   vector<int> tempRes;
                   int sum = nums[i] + nums[j] + nums[pre] + nums[tail];
                   if (sum == target)
                   {
                       tempRes.push_back(nums[i]);
                       tempRes.push_back(nums[j]);
                       tempRes.push_back(nums[pre]);
                       tempRes.push_back(nums[tail]);
                       res2.insert(tempRes);
                       pre++;
                       tail--;
                   }
                   else if (sum < target)
                   {
                       pre++;
                   }
                   else
                       tail--;
               }
            }
        }
        for (auto r : res2)
        {
            res.push_back(r);
        }
        return res;
    }
};