0
点赞
收藏
分享

微信扫一扫

三数之和双指针法

版本一:

public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if (nums == null) {
return res;
}

Arrays.sort(nums);
int len = nums.length;
for(int i = 0; i < len; i++) {
// i之后的数全部大于0
if (nums[i] > 0) {
return res;
}

if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}

int left = i + 1;
int right = len - 1;

while (left < right) {
int sum = nums[i] + nums[left] + nums[right];
if (sum < 0) {
left++;
// 当前的和前面一个
while(left < right && nums[left] == nums[left - 1]) left++;
} else if (sum > 0) {
right--;
// 当前的和后面一个
while(left < right && nums[right] == nums[right + 1]) right--;

} else {
res.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--;
}
}

}

return res;
}

版本二

public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if (nums == null) {
return res;
}

Arrays.sort(nums);
int len = nums.length;
for(int i = 0; i < len; i++) {
// i之后的数全部大于0
if (nums[i] > 0) {
return res;
}

if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}

int left = i + 1;
int right = len - 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++;
right--;
// 这两个去重逻辑和上面单独去重不一样
while(left < right && nums[left] == nums[left + 1]) left++;
while(left < right && nums[right] == nums[right - 1]) right--;

} else if (sum > 0) {
right--;
} else {
left++;
}

}

}

return res;
}


举报

相关推荐

0 条评论