4sum

原创于 2017-05-23 16:12:45 发布 · 308 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

leetcode解题思路专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种解决四数之和问题的高效算法。通过先排序数组，再利用双指针技巧寻找四个数相加等于目标值的组合，并通过去重操作确保结果的唯一性。

算法思路：步骤一：判断数组中元素个数，然后对数组排序；

步骤二：遍历数组中任意的两个数，然后设置left，right标志，如果sum<target 则left右移，否则，right左移，直到 sum == target；

步骤三：将sum==target的四个数构成的容器，存到vector<vector<int>> result 中；

步骤四：对容器中相同的元素进行去重：

1）sort(result.begin(),result.end()); 使用函数sort 对它进行排序

2) auto it =unique(result.begin(),result.end()); 使用函数unique对数组进行重新排列，“消除” 相邻的重复项（此刻的消除实际只是把重复的项放到的容器末尾），unique函数返回一个迭代器，指向最后一个不重复元素之后的位置；

3）erase（it,result.end()）;使用erase函数删除迭代器it之后的所有元素；容器中剩余部分的元素是不重复的。

注意：和3sum相比，思路基本一样，唯一的区别就是要多加一层for循环。

vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> result;
int n = nums.size();
sort(nums.begin(),nums.end());
if(n<4){
return result;
}
else{
for(int i=0;i<n-1;++i){
for(int j=i+1;j<n;++j){
int left=j+1;
   int right=n-1;
   while(left<right){
int sum=nums[i]+nums[j]+nums[left]+nums[right];
  if(sum == target){
vector<int> temp;
   temp.push_back(nums[i]);
temp.push_back(nums[j]);
temp.push_back(nums[left]);
temp.push_back(nums[right]);
result.push_back(temp);
left++;
    right--;
}
else if(sum<target){
left++;
}
else
right--;
}
}
   }
}
sort(result.begin(),result.end());
auto it=unique(result.begin(),result.end());
result.erase(it,result.end());
return result;
}