LeetCode 15. 三数之和

题目描述

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

示例：

  
给定数组 nums = [-1, 0, 1, 2, -1, -4]，  
  
满足要求的三元组集合为：  
[  
[-1, 0, 1],  
[-1, -1, 2]  
]  
  

解题思路

思路

这道题的思路是先排序，然后固定一个数，再用双指针去找另外两个数，时间复杂度为O(n^2)。

代码

class Solution {  
public List<List<Integer>> threeSum(int[] nums) {  
List<List<Integer>> res = new ArrayList<>();  
if (nums == null || nums.length < 3) {  
return res;  
}  
Arrays.sort(nums);  
for (int i = 0; i < nums.length - 2; i++) {  
if (i > 0 && nums[i] == nums[i - 1]) {  
continue;  
}  
int left = i + 1;  
int right = nums.length - 1;  
while (left < right) {  
int sum = nums[i] + nums[left] + nums[right];  
if (sum == 0) {  
res.add(Arrays.asList(nums[i], nums[left], nums[right]));  
while (left < right && nums[left] == nums[left + 1]) {  
left++;  
}  
while (left < right && nums[right] == nums[right - 1]) {  
right--;  
}  
left++;  
right--;  
} else if (sum < 0) {  
left++;  
} else {  
right--;  
}  
}  
}  
return res;  
}  
}  

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

解题思路

思路

这道题和三数之和的思路是一样的，只不过多了一层循环，时间复杂度为O(n^3)。

代码

class Solution {  
public List<List<Integer>> fourSum(int[] nums, int target) {  
List<List<Integer>> res = new ArrayList<>();  
if (nums == null || nums.length < 4) {  
return res;  
}  
Arrays.sort(nums);  
for (int i = 0; i < nums.length - 3; i++) {  
if (i > 0 && nums[i] == nums[i - 1]) {  
continue;  
}  
for (int j = i + 1; j < nums.length - 2; j++) {  
if (j > i + 1 && nums[j] == nums[j - 1]) {  
continue;  
}  
int left = j + 1;  
int right = nums.length - 1;  
while (left < right) {  
int sum = nums[i] + nums[j] + nums[left] + nums[right];  
if (sum == target) {  
res.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));  
while (left < right && nums[left] == nums[left + 1]) {  
left++;  
}  
while (left < right && nums[right] == nums[right - 1]) {  
right--;  
}  
left++;  
right--;  
} else if (sum < target) {  
left++;  
} else {  
right--;  
}  
}  
}  
}  
return res;  
}  
}