[leetcode] 18. 4Sum

来源:互联网 发布:sql 身份证号 编辑:程序博客网 时间:2024/06/15 20:53

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;    }};
0 0
原创粉丝点击