题目描述:
给定一个包含 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]
]
题目解答:
方法1:双指针
采用和三数之和相同的方法,先排序,然后外边是两层for循环,里边用双指针。注意需要跳过重复元素。
在该题中有一些优化措施,比如在两层循环之内,前两个数字a, b
已经确定,现在我们选择最后两个数字c, d
(加上前两个数字a, b
)来组成目标四个数字,此时如果比target
还小,说明即使后两个数字我们选择最大的,也不能等于target
,所以前两个数字a, b
选的太小了,所以可以直接增大第二个数字b
;如果现在我们选择第二个数字b
后边紧接着的两个数字c, d
(加上前两个数字a, b
)来组成目标四个数字,此时如果比target
大,则无论如何也不会等于target
,即前两个数字选的太大了,所以可以直接跳出,增大第一个数,第二个数字重新开始。
另外我先过滤了一些不合题意的数字,包括排序后数组最前边的和最后边的。比如最前边的,先假设我们选用最后三个最大的数字,此时加上最前边的数字a
,其和比target
还小,说明a
这个数字太小,无效,需要向后挪动。
运行时间4ms,代码如下。
/**
* Return an array of arrays of size *returnSize.
* Note: The returned array must be malloced, assume caller calls free().
*/
int comp(const void* a, const void* b) {
return (*(int*)a > *(int*)b ? 1 : -1);
}
int** fourSum(int* nums, int numsSize, int target, int* returnSize) {
int n = numsSize, i = 0, j = 0;
if(n < 4)
return NULL;
qsort(nums, n, sizeof(int), comp);
int start = 0, end = 0;
int temp = nums[n - 1] + nums[n - 2] + nums[n - 3];
for(start = 0; start < n - 3 && nums[start] + temp < target; start++) {
}
if(start == n - 3)
return NULL;
temp = nums[0] + nums[1] + nums[2];
for(end = n - 1; end > 2 && nums[end] + temp > target