1. 题目
给你一个由
n
个整数组成的数组nums
,和一个目标值target
。请你找出并返回满足下述全部条件且不重复的四元组[nums[a], nums[b], nums[c], nums[d]]
(若两个四元组元素一一对应,则认为两个四元组重复):
0 <= a, b, c, d < n
a
、b
、c
和d
互不相同nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。
2. 示例
3. 分析
做这题之前先做这道:三数之和,对应题解:三数之和 - 题解
四数之和无非就是再多套一层循环,即再增加一个固定数。之后就利用双指针寻找 两数之和 == target - 第一个固定数 - 第二个固定数:
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
sort(nums.begin(), nums.end());
vector<vector<int>> ret;
int n = nums.size();
for(int i = 0; i < n;) // 第一个固定数
{
for(int j = i + 1; j < n;) // 第二个固定数
{
// 双指针
int left = j + 1, right = n - 1;
long long aim = (long long)target - nums[i] - nums[j];
while(left < right)
{
int sum = nums[left] + nums[right];
if(sum > aim) right--;
else if(sum < aim) left++;
else
{
ret.push_back({nums[i], nums[j], nums[left++], nums[right--]});
while(left < right && nums[left] == nums[left-1]) left++; // 去重左指针元素
while(right < right && nums[right] == nums[right+1]) right--; // 去重右指针元素
}
}
j++;
while(j < n && nums[j] == nums[j-1]) j++; // 去重第二个固定数指针元素
}
i++;
while(i < n && nums[i] == nums[i-1]) i++; // 去重第一个固定数指针元素
}
return ret;
}
};