LeetCode18:四數之和(Medium)
阿新 • • 發佈:2018-12-13
題目描述:
給定一個包含 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] ]
這個跟三數之和的思維一樣的,就是多添加了個for迴圈,用了四個指標
c++實現:
class Solution { public: vector<vector<int>> fourSum(vector<int>& nums, int target) { sort(nums.begin(),nums.end()); vector<int> sum(4); vector<vector<int> > res; int len=nums.size(); int tsum; int i,j,l,k; for (i=0;i<len-3;i++) { if(i>0&&nums[i]==nums[i-1]) continue; for(j=i+1;j<len-2;j++){ if(j>i+1 && nums[j]==nums[j-1]) continue; for(l=j+1,k=len-1;l<k;){ tsum=nums[i]+nums[j]+nums[l]+nums[k]; if(tsum==target){ sum[0]=nums[i]; sum[1]=nums[j]; sum[2]=nums[l]; sum[3]=nums[k]; res.push_back(sum); while(l<k && nums[l]==nums[l+1]) l++; while(l<k && nums[k]==nums[k-1]) k--; l++; k--; } else if (tsum<target) l++; else k--; } } } return res; } };