标签:个数 目标 相等 移动 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 };
标签:个数 目标 相等 移动 mil 4sum 不可 turn return
原文地址:https://www.cnblogs.com/wmx24/p/9205532.html