public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
//1. 先对数组进行排序:此处可以使用排序算法:快速排序、归并排序。
Arrays.sort(nums);
//2. 双指针,一个从前面找,一个从后面找
for (int i = 0; i < nums.length; i++) {
//排序之后如果第一个元素已经大于零,那么无论如何组合都不可能凑成三元组,直接返回结果就可以了
if (nums[i] > 0) {
return res;
}
//减枝操作:第一个元素去重;(i > 0为了防止报索引异常)
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];
if (sum == 0) {
res.add(Arrays.asList(nums[i], nums[left], nums[right]));
//跳过所有重复的值
while (left < right && nums[right] == nums[right - 1]) {
right--;
}
while (left < right && nums[left] == nums[left + 1]) {
left++;
}
//找到答案,指正同时收缩
right--;
left++;
} else if (sum > 0) {
right--;
} else {
left++;
}
}
}
return res;
}
三数之和:(排序+双指针+剪枝)
最新推荐文章于 2024-09-12 18:58:16 发布