描述
给出一个整数数组, 数组中是否存在任意 3 个数 a, b, c 满足 a + b + c = 0? 找出数组中所有满足以上条件的三元组,最后输出这些三元组的个数(包含相同元素的三元组只计算一次)。
输入
一个包含多个整数(正或负)的字符串,每个整数之间用逗号(,)分隔,如:-1,0,1,2,-1,-4。
输出
输入满足加和结果正好等于 0 的三元组的个数,如对于 -1,0,1,2,-1,-4 有 [-1, 0, 1] 和 [-1, -1, 2],所以输出 2
输入样例
-1,0,1,2,-1,-4
复制样例
输出样例
2
第一次使用容器,A这道题比较艰辛,主要用到了set容器,其他不多说,直接上代码,尽管简单,但仍值得我思考的一个问题
package 小米oj;
import java.util.ArrayList;
import java.util.Arrays;
public class 和为零的三元组 {
public static void main(String[] args) {
System.out.println(solution("-1,0,1,2,-1,-4"));
}
private static String solution(String line) {
String a[] = line.split(",");
int a1[] = new int[a.length];
java.util.Set<java.util.List<Integer>> set = new java.util.HashSet<>();
for (int i = 0; i < a.length; i++) {
a1[i] = Integer.parseInt(a[i]);
}
Arrays.sort(a1);
int sum = 0;
for (int i = 0; i < a1.length - 2; i++) {
int high = a1.length - 1;
int low = i + 1;
while (low < high) {
if (a1[i] + a1[low] + a1[high] < 0)
low++;
else if (a1[i] + a1[low] + a1[high] > 0)
high--;
if (a1[i] + a1[low] + a1[high] == 0&&low!=high) {
java.util.List<Integer> t1 = new java.util.ArrayList<>();
t1.add(a1[i]);
t1.add(a1[low]);
t1.add(a1[high]);
set.add(t1);
low++;
}
}
}
return String.valueOf(set.size());
}
}