给定一个包含 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] ]
看过三数之和的同学,不妨拿这道题练练手。
方法一:双指针法
思路:参考三数之和的思想。时间复杂度O(n^2),空间复杂度O(logn)
【1】先对数组进行排序
【2】对数组进行第一遍 外层遍历,其值为a。nums[i]=a
【3】对a之后的数组进行遍历,其值为b。nums[j]=b。(j=i+1)
【4】对b之后的数组进行双指针遍历,其中c指向j+1,d指向数组尾节点。四数之和为sum,sum=a+b+c+d。并与target进行比较 【5】若大于target,d进行左移以降低sum。若小于target,c进行右移以增加sum。否则符合条件添加vector。
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int>> ans;
if (nums.size() <= 3)
return ans;
sort(nums.begin(), nums.end()); //排序
for (int i = 0;i <= nums.size() - 4;++i) {
if (i > 0 && nums[i - 1] == nums[i]) //第一次去重
continue;
int a = nums[i];
for (int j = i + 1;j <= nums.size() - 3;++j) {
if (j > i + 1 && nums[j - 1] == nums[j]) //第二次去重
continue;
int b = nums[j], begin = j + 1, end = nums.size() - 1;
while (begin < end) {
int sum = a + b + nums[begin] + nums[end]; //a+b+c+d
if (sum == target) {
vector<int> vec = { a,b,nums[begin],nums[end] };
ans.push_back(vec);
while (begin < end&&nums[begin] == nums[begin + 1]) //第三次去重
++begin;
++begin;
while (begin < end&&nums[end] == nums[end - 1])
--end;
--end;
}
else if (sum < target) { //a+b+c+d<target,增大c
while (begin < end&&nums[begin] == nums[begin + 1])
++begin;
++begin;
}
else { //a+b+c+d>target,减小d
while (begin < end&&nums[end] == nums[end - 1])
--end;
--end;
}
}
}
}
return ans;
}