Leetcode 18.四数之和
题目
给定一个包含 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]
]
题解
此题与三数之和的核心思想相同,利用二分查找确定两个数。因此,此题中我们需要二次循环确定两个数,然后再利用二分法查找剩下的两个数。
我们首先对数组进行排序。外层循环确定最大的数,内层循环确定最小的数,然后利用二分查找的思想确定中间的两个数。即我们假设a为最小数,b为次小数,令a=nums[i],d=nums[j],其中i<j,之后令x=j+1,y=n-1,nums[x]、nums[y]分别作为c、d。如果此时的和大于target,则y–;此时的和小于target,则x++。为了避免重复的四元组出现,我们要把a、b、c、d可能重复取的值略掉,详细方法见代码
代码
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> ans;
sort(nums.begin(),nums.end());
int n=nums.size(),i,j,y,x;
for(i=0; i<n-3; ){
for(j=i+1; j<n-2; ){
x=j+1; y=n-1;
while(x < y){
int now = nums[i]+nums[j]+nums[x]+nums[y];
if(now == target){
if(x == j+1 || nums[x]!=nums[x-1]){
vector<int> num = {nums[i],nums[j],nums[x],nums[y]};
ans.push_back(num);
}
do{x++;}while(nums[x]==nums[x-1] && x<y); //避免c取重复的值
do{y--;}while(nums[y]==nums[y+1] && x<y); //避免d取重复的值
}else if(now > target){
do{y--;}while(nums[y]==nums[y+1] && x<y); //可以用y--;一句代替
}else{
do{x++;}while(nums[x]==nums[x-1] && x<y); //可以用x++;一句代替
}
}
do{j++;}while(nums[j-1] == nums[j] && j<n-2); //避免b取重复的值
}
do{i++;}while(nums[i-1] == nums[i] && i<n-3); //避免a取重复的值
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。