給出一個有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;
}