给出一个有n个元素的数组S,S中是否有元素a,b,c满足a+b+c=0?找出数组S中所有满足条件的三元组。
注意:
三元组(a、b、c)中的元素必须按非降序排列。(即a≤b≤c)
解集中不能包含重复的三元组。
例如,给定的数组 S = {-10 0 10 20 -10 -40},解集为(-10, -10, 20),(-10, 0, 10)
0 <= S.length <= 1000
示例1
输入:[0]
返回值:[]
示例2
输入:[-2,0,1,1,2]
返回值:[[-2,0,2],[-2,1,1]]
示例3
输入:[-10,0,10,20,-10,-40]
返回值:[[-10,-10,20],[-10,0,10]]
public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
if (num == null) {
return result;
}
//排序
Arrays.sort(num);
int sum, left, right;
for (int i = 0; i < num.length - 2; i++) {
//避免重复计算
if (i != 0 && num[i] == num[i - 1]) {
continue;
}
left = i + 1;
right = num.length - 1;
/**
* 固定一个数,从后面的数中选出两个数,因为数组是有序的,所以可以
* 用两个数组下标left和right,left指向当前元素的后一个位置,right指向数组的最后一个位置
* 三个数相加的和等于0时,加入解集;
* 小于0时,把left往右边移动;
* 大于0时,把right往左边移动;
*/
while (left < right) {
sum = num[left] + num[right];
if (sum + num[i] == 0) {
ArrayList<Integer> solution = new ArrayList<>();
solution.add(num[i]);
solution.add(num[left]);
solution.add(num[right]);
result.add(solution);
left++;
right--;
//跳过重复元素计算
while (left < right && num[left] == num[left - 1]) {
left++;
}
//跳过重复元素计算
while (left < right && num[right] == num[right + 1]) {
right--;
}
} else if (sum + num[i] < 0) {
left++;
} else {
right--;
}
}
}
return result;
}