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--;
}
浙公网安备 33010602011771号