题解 | #数组中相加和为0的三元组#
数组中相加和为0的三元组
http://www.nowcoder.com/practice/345e2ed5f81d4017bbb8cc6055b0b711
- 左右指针分别在两侧,向中心靠拢
- 使用set去重
- 如果等于0,在当前i循环下,不应该直接break,可能存在其他的组合,所有l++,r--,继续遍历可能存在的组合
- 返回的list严格遵守升序,所以要进行一次排序。
import java.util.*;
public class Solution {
public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
Arrays.sort(num);
Set<ArrayList<Integer>> set = new HashSet<>();
for (int i = 1; i < num.length - 1; i++) {
int begin = 0;
int end = num.length - 1;
while(begin < i && i < end) {
int sum = num[i] + num[begin] + num[end];
if (sum == 0) {
ArrayList<Integer> tmp = new ArrayList<Integer>();
tmp.add(num[begin]);
tmp.add(num[i]);
tmp.add(num[end]);
set.add(tmp);
begin++;
end--;
}else if (sum > 0) {
end--;
}else {
begin++;
}
}
}
ArrayList<ArrayList<Integer>> list = new ArrayList<>(set);
list.sort((o1, o2) -> {
for (int i = 0; i < o1.size(); i++) {
if (o1.get(i) == o2.get(i)) {
continue;
}
if (o1.get(i) < o2.get(i)) {
return -1;
}else {
return 1;
}
}
return 0;
});
return list;
}
}