1. 题目
https://leetcode.cn/problems/3sum/
2. 解法
思路是这样的:
- 首先,对数组进行排序,这样可以方便地跳过重复的元素,也可以利用双指针法来寻找第二个数和第三个数。
- 然后,遍历数组,固定第一个数,如果第一个数大于零,那么后面的数都大于零,不可能有解,所以可以直接结束循环。
- 接着,使用双指针法,从左右两端向中间寻找第二个数和第三个数,如果三数之和为零,就找到一个解,并跳过重复的元素;如果三数之和小于零,就说明第二个数太小,需要增大;如果三数之和大于零,就说明第三个数太大,需要减小。
- 最后,返回结果列表。
第一元素固定+后两个元素采用双指针
public List<List<Integer>> threeSum(int[] nums) { //先对数组进行排序 Arrays.sort(nums); //创建一个结果列表 List<List<Integer>> result = new ArrayList<>(); //遍历数组,从第一个数开始,到倒数第三个数结束 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]; //如果和为0,就找到了一个解 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) { //如果和大于0,说明右边的数太大了,需要减小,所以右指针左移 right--; } else { //如果和小于0,说明左边的数太小了,需要增大,所以左指针右移 left++; } } } //返回结果列表 return result; }
其中以下三处 处理重复 非常重要
//如果当前数和前一个数相同,就跳过,避免重复
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
//左指针右移,跳过相同的数,避免重复
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
//右指针左移,跳过相同的数,避免重复
while (left < right && nums[right] == nums[right - 1]) {
right--;
}