目錄
- 需要注意的幾點
- 1、去除剪枝操作
- 2、去重操作的細節
- code以及效果:
題目
給定一個包含 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、去除剪枝操作
之前三個元素相加我們有個剪枝操作:if(nums[j] > 0) return result;
此時我們仍然這樣想:如果第一個數>target,因為是遞增的,所以之后的序列相加必定大于target(其實這個是錯的,如果是負數只會越來越小)
例如:
[1,-2,-5,-4,-3,3,3,5]
-11
之前之所以可以剪枝是限制了target=0,這樣就能保證相加最起碼遞增的。
2、去重操作的細節
if(j>0 && nums[j]==nums[j-1]) continue;if(i>j+1 && nums[i]==nums[i-1]) continue;
如果是[0,0,0,0],target=0;(一開始這邊報錯,然后修改了去重操作的條件)
探討一下去重過程:
j = 0;i = 1;left=2;right=3;此時不滿足去重,然后在for循環中完成結果。
j = 0;i = 2;去重。
…
code以及效果:
class Solution {
public:vector<vector<int>> fourSum(vector<int>& nums, int target) {vector<vector<int>> result;//對nums排序sort(nums.begin(), nums.end());int n = nums.size();for(int j=0;j<n;j++){if(j>0 && nums[j]==nums[j-1]) continue;//如果第一個數>target,因為是遞增的,所以之后的序列相加必定大于target(其實這個是錯的,如果是負數只會越來越小)//if(nums[j] > target) return result;for(int i=j+1;i<n;i++){//使用雙指針法,不過得先確定一個數(因為是3個數),這里我們確定第一個數//接下來是去除重復元素//如果是[0,0,0,0],target=0時,去重可能會把答案去除掉if(i>j+1 && nums[i]==nums[i-1]) continue;//接下來就是雙指針操作int left = i+1;int right = n-1;while(left < right){int sum =nums[j] + nums[i] + nums[left] + nums[right];if(sum > target){//說明right太大right--;}else if(sum < target){//說明left太小left++;}else{result.emplace_back(vector<int>{nums[j],nums[i],nums[left],nums[right]});//找到元素之后,left++,right--(為什么right--?因為nums[i],nums[left],nums[right]==0,//此時left++,且由于遞增,nums[left++]>nums[left],為了能夠再次找到sum==0的數對,nums[new_right]必須//left去重往右靠while(left < right && nums[left]==nums[left+1]) left++;//right去重往左靠while(left < right && nums[right]==nums[right-1]) right--;//小于nums[old_right])left++;right--;}}}}return result;}
};
還可以通過其他技巧進行求解,此處不做過度探討。等以后整理到相關題型的時候再做修改以及更新。
回溯法求解