Nameless Site

But one day, you will stand before its decrepit gate,without really knowing why.

0%

四数之和

来源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]
]

这题思路其实和之前的三数之和是一样的,但是由于我自己逻辑不够完善,导致了卡在这题很久。
同样是采用双指针,先固定2个数,在对剩下2层进行遍历操作检查sum是否和target相等。
在最外层遍历时,受到了三数之和这题的影响,判断了第一层if nums[s] > target ,then break;
这其实是没有必要的,因为有负数的存在,当有负数时,连续几个负数相加会越来越小,从而满足target。
其次,在第二层遍历时,同样的,if nums[j] > target,then break;
这也是没必要的,原因同上。
最后附上完成代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> ans = new ArrayList();
int len = nums.length;
if(nums == null || len < 4) return ans;
Arrays.sort(nums); // 排序
for (int i = 0; i < len - 3; i++) {
//if(nums[i] > target) break; // 如果当前数字大于target,则4数之和一定大于target,所以结束循环
if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
if((nums[i] + nums[i+1] + nums[i+2] + nums[i+3])>target)
break;
if((nums[i]+nums[len-1]+nums[len-2]+nums[len-3])<target)
continue;
for(int j = i + 1;j<len - 2;j++)
{
if(j > i+1 && nums[j] == nums[j-1]) continue; // 去重
//if(nums[j] > target) break; // 如果当前数字大于target,则4数之和一定大于target,所以结束循环
//if(nums[i] + nums[j] > target) break;
if((nums[i] + nums[j] + nums[len - 1] + nums[len - 2]) < target) //最大4个数字组合都小于target,进入下一个数字的循环
continue;
if((nums[i] + nums[j] + nums[j+1] + nums[j+2])>target)
break;

int L = j+1;
int R = len-1;

while(L < R){
int sum = nums[i] + nums[j] + nums[L] + nums[R];
if(sum == target){
ans.add(Arrays.asList(nums[i],nums[j],nums[L],nums[R]));
while (L<R && nums[L] == nums[L+1]) L++; // 去重
while (L<R && nums[R] == nums[R-1]) R--; // 去重
L++;
R--;
}
else if (sum < target)
while(L<R&&nums[L]==nums[++L]);
else if (sum > target)
while(L<R&&nums[R]==nums[--R]);
}
}
}
return ans;
}
}