力扣Hot100题目解析:15. 三数之和(Java详解)
一、题目描述
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a、b、c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
示例:
输入: nums = [-1,0,1,2,-1,-4]
输出: [[-1,-1,2],[-1,0,1]]
二、解题思路(排序+双指针)
本题的核心在于如何高效地查找三个数相加等于0且不重复。
步骤解析:
- 先对数组进行排序,这样方便后续去重和处理。
- 固定第一个数 nums[i],然后用两个指针 left 和 right 分别指向 i+1 和数组末尾。
- 移动两个指针,计算三数之和 sum:
- 如果 sum == 0,说明找到一组解,加入结果,并分别跳过重复元素(去重)。
- 如果 sum < 0,说明和小了,需要 left++;如果 sum > 0,说明和大了,需要 right--。
- 注意所有数都不能重复取,外层和内层都需要去重。
这样可以将时间复杂度降为 O(n^2)。
三、Java代码实现(详细注释)
import java.util.*;
public class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(nums); // 1. 排序
int n = nums.length;
for (int i = 0; i < n - 2; i++) {
// 2. 去重:跳过重复的固定数
if (i > 0 && nums[i] == nums[i - 1]) continue;
int left = i + 1, right = n - 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]));
// 跳过重复的left
while (left < right && nums[left] == nums[left + 1]) left++;
// 跳过重复的right
while (left < right && nums[right] == nums[right - 1]) right--;
// 移动指针
left++;
right--;
} else if (sum < 0) {
left++; // 和小了,增大left
} else {
right--; // 和大了,减小right
}
}
}
return res;
}
}
代码详解:
Arrays.sort(nums)
:排序方便双指针和去重。- 外层循环固定一个数,内层双指针查找剩下两个数。
- 每次找到解后,要跳过所有重复的left和right,避免重复解。
- 时间复杂度O(n^2),空间复杂度O(1)(不计结果)。
四、总结
三数之和是典型的排序+双指针题目,学会了本题可以举一反三解决多数k-sum问题。去重细节和指针移动要特别注意。