LeetCode 18. 四数之和（4Sum）

Posted 2020-11-19 FlyingWarrior

 

题目描述

 

给定一个包含 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 };