15. 三数之和
思路
双指针
图解+思路
首先先对数组进行排序。排序后创建三个指针 如下图
核心思想:(如果不好理解可以反着想一下)
三数之和小于0,left指针向右移
三数之和大于0,right向左移
等于0的时候,返回答案之后要立即去重!
细节:
题目要求没有答案返回空集合,所以要考虑给的数组是否合规
去重 时要注意:每次i移动都要给i指针去重;每次找到答案以后,要给left和right去重
代码(附有详细注释)
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
ArrayList<List<Integer>> result = new ArrayList<>(16);
//如果数组长度小于3直接返回空集合(题目说明)
if (nums.length < 3) {
return result;
}
//排序
Arrays.sort(nums);
//遍历到最后i指针的后面也会有left和right两个指针,所以遍历次数是nums长度-2
for (int i = 0; i < nums.length - 2; i++) {
int left = i + 1;
int right = nums.length - 1;
//如果当前元素大于0,三数之和一定大于0
if(nums[i] > 0){
break;
}
//i指针去重
if (i > 0 && nums[i] == nums[i - 1]){
continue;
}
while (left < right){
int sum = nums[i] + nums[left] + nums[right];
//找到了!
if ( sum == 0){
result.add(new ArrayList<>(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--;
//三数之合小于0,左指针向右移动
}else if (sum < 0){
left ++;
//反之,右指针向左移动
}else if (sum > 0){
right --;
}
}
}
return result;
}
}