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] ]
思路:
嗯......和上一题三指针如出一辙。只需在上一题三数和的做法上再加一层循环即可。当然,这样的话因为多加了一层循环就需要多进行一个数的去重,不过去重的思路是完全一样的。,
重要的来了,本题与上一题三数和有一个不一样的地方在于target可能会小于0!而上一题是求解三数和等于0,而在三数和等于0的题目里面可以使用break语句剪枝。
if (nums[i] > 0) {
break;
}
本题不能使用 if (nums[i] > target) break;进行剪枝,因为当target小于0的情况下会出错(自己设置一组target < 0的样例模拟走一遍算法流程即可明白)。
参考代码:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
vector<vector<int> > result;
int len = nums.size();
for (int i = 0; i < len; i++) {
if (i > 0 && nums[i] == nums[i -1]) {
continue;
}
for (int j = i + 1; j < len; j++) {
if (j > i + 1 && nums[j] == nums[j -1]) {
continue;
}
int left = j + 1, right = len - 1;
while (left < right) {
if (nums[left] + nums[right] > target - nums[i] - nums[j]) {
right--;
}
else if (nums[left] + nums[right] < target - nums[i] - nums[j]) {
left++;
}
else {
result.push_back({nums[i], nums[j], nums[left], nums[right]});
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
right--;
left++;
}
}
}
}
return result;
}
};