avatar

day3四数之和

先放代码了,后面有时间会在整理下。最近事情挺多的

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
//给定一个包含 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]
//]
//
// Related Topics 数组 哈希表 双指针


import java.util.ArrayList;
import java.util.Arrays;

//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public List<List<Integer>> fourSum(int[] nums, int target) {
int len = nums.length;
List<List<Integer>> ans = new ArrayList<>();
if (len < 4) {
return ans;
}
// 对数组排序,方便遍历和去重。
Arrays.sort(nums);
// 当目标值比数组最小四数和还小或者比数组最大四数和还大,表示没有符合条件的四个数。
if (target < nums[0] + nums[1] + nums[2] + nums[3]
|| target > nums[len - 1] + nums[len - 2] + nums[len - 3] + nums[len - 4]) {
return ans;
}
// 循环固定第一个数,然后循环遍历选择另外三个数。
for (int i = 0; i < len - 3; i++) {
// 相同的第一个数只固定一次,避免重复运算。
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
// 如果 target 小于当前循环能得到的最小和,则跳出循环。
if (target < nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3]) {
break;
}
// 如果 target 大于此轮循环的最大和,则继续循环下一个比较大的第一位数。
if (target > nums[i] + nums[len - 1] + nums[len - 2] + nums[len - 3]) {
continue;
}
// 循环固定第一个数,根据双指针从它之后去选择另外两个数。
for (int j = i + 1; j < len - 2; j++) {
// 相同的第一个数只固定一次,避免重复运算。
if (j > i + 1 && nums[j] == nums[j - 1]) {
continue;
}
// 如果 target 小于当前循环能得到的最小四数和,则跳出循环。
if (target < nums[i] + nums[j] + nums[j + 1] + nums[j + 2]) {
break;
}
// 如果 target 大于此轮循环的最大和,则继续循环下一个比较大的第二位数。
if (target > nums[i] + nums[j] + nums[len - 1] + nums[len - 2]) {
continue;
}
int L = j + 1, R = len - 1;
while (L < R) {
int sum = nums[i] + nums[j] + nums[L] + nums[R];
// 根据四数之和与 target 大小的比较来移动两指针。
if (sum == target) {
ans.add(Arrays.asList(nums[i], nums[j], nums[L], nums[R]));
L++;
while (L < R && nums[L - 1] == nums[L]) {
L++;
}
R--;
while (L < R && nums[R + 1] == nums[R]) {
R--;
}
} else if (sum > target) {
R--;
while (L < R && nums[R + 1] == nums[R]) {
R--;
}
} else {
L++;
while (L < R && nums[L - 1] == nums[L]) {
L++;
}
}
}
}
}
return ans;
}

}
//leetcode submit region end(Prohibit modification and deletion)
文章作者: 无知的小狼
文章链接: https://bytedance.press/2020/04/27/20200401/title%20day3%E5%9B%9B%E6%95%B0%E4%B9%8B%E5%92%8C/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 无知的小狼

评论