三数之和
三数之和(3Sum)要求在给定的整数数组中,找出所有满足三个数相加等于目标值的三元组,且结果不能包含重复的三元组。
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class ThreeSum {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new ArrayList<>();
// 首先对数组进行排序
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) {
// 如果找到了满足条件的三元组,将其添加到结果列表中
result.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 result;
}
}
解释每一步:
-
首先对数组进行排序,这样可以方便后续使用双指针法进行遍历。
-
遍历数组,将三数之和问题转化为两数之和问题。在遍历的过程中,每次固定一个数
nums[i]
,并在其后面的范围内使用双指针法找到两数之和等于目标值的组合。 -
使用双指针法,左指针
left
指向当前元素的下一个位置,右指针right
指向数组的最后一个元素。 -
如果三数之和等于目标值,则将该三元组添加到结果列表中,并继续向内收缩指针以避免重复解。
-
如果和小于目标值,增大左指针;如果和大于目标值,减小右指针。
-
重复以上步骤直到遍历完成。
这种解法的时间复杂度为 O(n^2),其中 n 是数组的长度。因为我们首先对数组进行排序,排序的时间复杂度为 O(nlogn),然后使用双指针法在最坏情况下的时间复杂度为 O(n^2)。空间复杂度为 O(1),因为只使用了常数级别的额外空间。