码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode 18. 四数之和(4Sum)

时间:2018-06-20 21:22:10      阅读:124      评论:0      收藏:0      [点我收藏+]

标签:个数   目标   相等   移动   mil   4sum   不可   turn   return   

 

题目描述

 

给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

注意:

答案中不可以包含重复的四元组。

示例:

给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

满足要求的四元组集合为:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

 

解题思路

 

借鉴三数之和的思路,首先将数组从大到小排序,然后从数组首位置开始遍历:

  • 固定第一个数,将其加入结果集合,为了避免重复,若该数与前一个数相同则跳过
  • 接着固定第二个数为第一个数的下一个位置,将其加入结果集合,同样为了避免重复,若该数与前一个数相同则跳过
  • 用双指针法,左右指针分别指向第二个数的下一个位置和最后一个位置,算出两数的目标和,若两位置的数之和正好等于目标,则加入到结果集合中,并将左指针指向右边第一个与当前数不相同的位置,同时右指针向左移动一位;若两位置的数之和小于目标,则将左指针向右移动一位;若两位置的数之和大于目标,则将右指针向左移动一位。这样移动直到左指针超过右指针或与其相遇
  • 根据回溯法的思想,遍历完第一个数和第二个数的解之后,分别将其弹出结果集合

 

代码

 

 1 class Solution {
 2 public:
 3     vector<vector<int>> fourSum(vector<int>& nums, int target) {
 4         sort(nums.begin(), nums.end());
 5         vector<vector<int>> res;
 6         vector<int> temp;
 7         int size = nums.size();
 8         for(int i = 0; i < size - 3; i++){
 9             if(i && nums[i] == nums[i - 1]) continue;
10             int sum3 = target - nums[i];
11             temp.push_back(nums[i]);
12             for(int j = i + 1; j < size - 2; j++){
13                 if(j != i + 1 && nums[j] == nums[j - 1]) continue;
14                 int sum2 = sum3 - nums[j];
15                 temp.push_back(nums[j]);
16                 int left = j + 1, right = size - 1;
17                 while(left < right){
18                     if(nums[left] + nums[right] == sum2){
19                         temp.push_back(nums[left]);
20                         temp.push_back(nums[right]);
21                         res.push_back(temp);
22                         temp.pop_back();
23                         temp.pop_back();
24                         left++;
25                         while(nums[left] == nums[left - 1]) left++;
26                         right--;
27                     }
28                     else if(nums[left] + nums[right] < sum2)
29                         left++;
30                     else right--;
31                 }
32                 temp.pop_back();
33             }
34             temp.pop_back();
35         }
36         return res;
37     }
38 };

 

LeetCode 18. 四数之和(4Sum)

标签:个数   目标   相等   移动   mil   4sum   不可   turn   return   

原文地址:https://www.cnblogs.com/wmx24/p/9205532.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!