力扣四數(shù)之和

文章目錄

• 力扣四數(shù)之和
• • 一、題目描述
  • 二、分析
  • 三、代碼
  • 四、問題描述
  • 五、分析
  • 六、代碼
  • 七、100Sum 問題？

一、題目描述

二、分析

這道題和：三數(shù)之和：三數(shù)之和類似，直接放代碼：

三、代碼

class Solution { public:// 只需要返回三個數(shù)的和,根據(jù)15的雙指針來改int threeSumClosest(vector<int>& nums, int target) {int n = nums.size();if( n < 3 )return NULL;if( n == 3 )return nums[0]+nums[1] + nums[2];// 仍然需要重排sort( nums.begin(), nums.end() );int minMerge = INT_MAX;int ans = 0;int tmpSum = 0;for( int first = 0; first < n; first++ ){int second = first + 1;int third = n - 1;while( second < third ){tmpSum = nums[first] + nums[second] + nums[third];// 如果當(dāng)前距離小于目前最小距離，則需要更新if( abs(tmpSum - target) < minMerge ){ minMerge = abs( tmpSum - target );ans = tmpSum;}// 然后決定下一步怎么走if( tmpSum < target ){ // 首先，tmpSum < target說明右側(cè)可能有更好的結(jié)果，必須是像右側(cè)走，second++second++;}else if (tmpSum > target){third--;}else{ // 如果已經(jīng)相等，那么間隔就是0return ans;}}}return ans;} };

四、問題描述

五、分析

原理和三數(shù)之和完全一樣

六、代碼

class Solution{public: vector<vector<int>> fourSum(vector<int>& nums, int target) {sort(nums.begin(),nums.end());vector<vector<int> > res;if(nums.size() < 4)return res;int a,b,c,d,_size=nums.size();for(a = 0;a <= _size - 4;a++){if(a > 0 && nums[a] == nums[a - 1]) continue; //確保nums[a] 改變了for(b = a + 1;b <= _size - 3;b++){if(b > a + 1 && nums[b] == nums[b - 1])continue; //確保nums[b] 改變了c = b + 1,d = _size - 1;while(c < d){if(nums[a] + nums[b] + nums[c] + nums[d] < target)c++;else if(nums[a] + nums[b] + nums[c] + nums[d] > target)d--;else{res.push_back({nums[a],nums[b],nums[c],nums[d]});while(c < d && nums[c + 1] == nums[c]) //確保nums[c] 改變了c++;while(c <d && nums[ d - 1] == nums[d]) //確保nums[d] 改變了d--;c++;d--;}}}}return res;} };

七、100Sum 問題？

• 在 LeetCode 上，4Sum 就到頭了，但是回想剛才寫 3Sum 和 4Sum的過程，實際上是遵循相同的模式的。我相信你只要稍微修改一下 4Sum 的函數(shù)就可以復(fù)用并解決 5Sum 問題，然后解決 6Sum問題……
• 那么，如果我讓你求 100Sum 問題，怎么辦呢？其實我們可以觀察上面這些解法，統(tǒng)一出一個 nSum 函數(shù)：

/* 注意：調(diào)用這個函數(shù)之前一定要先給 nums 排序 */ vector<vector<int>> nSumTarget(vector<int>& nums, int n, int start, int target) {int sz = nums.size();vector<vector<int>> res;// 至少是 2Sum，且數(shù)組大小不應(yīng)該小于 nif (n < 2 || sz < n) return res;// 2Sum 是 base caseif (n == 2) {// 雙指針那一套操作int lo = start, hi = sz - 1;while (lo < hi) {int sum = nums[lo] + nums[hi];int left = nums[lo], right = nums[hi];if (sum < target) {//去掉重復(fù)的結(jié)果while (lo < hi && nums[lo] == left) lo++;} else if (sum > target) {//去掉重復(fù)的結(jié)果while (lo < hi && nums[hi] == right) hi--;} else {res.push_back({left, right});while (lo < hi && nums[lo] == left) lo++;while (lo < hi && nums[hi] == right) hi--;}}} else {// n > 2 時，遞歸計算 (n-1)Sum 的結(jié)果for (int i = start; i < sz; i++) {vector<vector<int>> sub = nSumTarget(nums, n - 1, i + 1, target - nums[i]);for (vector<int>& arr : sub) {// (n-1)Sum 加上 nums[i] 就是 nSumarr.push_back(nums[i]);res.push_back(arr);}while (i < sz - 1 && nums[i] == nums[i + 1]) i++;}}return res; }

• 嗯，看起來很長，實際上就是把之前的題目解法合并起來了，n == 2 時是 twoSum 的雙指針解法，n > 2時就是窮舉第一個數(shù)字，然后遞歸調(diào)用計算 (n-1)Sum，組裝答案。
• 需要注意的是，調(diào)用這個 nSum 函數(shù)之前一定要先給 nums 數(shù)組排序，因為 nSum 是一個遞歸函數(shù)，如果在 nSum 函數(shù)里調(diào)用排序函數(shù)，那么每次遞歸都會進行沒有必要的排序，效率會非常低。
• 比如說現(xiàn)在我們寫 LeetCode 上的 4Sum 問題：

vector<vector<int>> fourSum(vector<int>& nums, int target) {sort(nums.begin(), nums.end());// n 為 4，從 nums[0] 開始計算和為 target 的四元組return nSumTarget(nums, 4, 0, target); }

• 再比如 LeetCode 的 3Sum 問題，找 target == 0 的三元組：

vector<vector<int>> threeSum(vector<int>& nums) {sort(nums.begin(), nums.end());// n 為 3，從 nums[0] 開始計算和為 0 的三元組return nSumTarget(nums, 3, 0, 0); }

那么，如果讓你計算 100Sum 問題，直接調(diào)用這個函數(shù)就完事兒了。

總結(jié)

以上是生活随笔為你收集整理的力扣四数之和的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。