版本一:
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;
}