序号:#15
难度:有挑战
时间限制:1000ms
内存限制:10M
描述
给出一个整数数组, 数组中是否存在任意 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
希望我的代码能被看懂吧
/**
* 已引入:
* java.util.*
* 要使用其他 jar 包请使用完整路径,如:
* java.util.List<java.util.List<Integer>> list = new java.util.ArrayList<java.util.List<Integer>>(100);
* @param line 为单行测试数据
* @return 处理后的结果
*/
private static String solution(String line) {
// please write your code here
String[] str = line.split(",");
int[] a = new int[str.length];
if(a.length<3) {
return String.valueOf(0);
}
for (int i = 0; i < str.length; i++) {
a[i] = Integer.parseInt(str[i]);
}
Arrays.sort(a);
int count = 0;
int headFlag=0,endFlag=0;
int fuck=0;
for(int i=0;i<a.length;i++) {
if(a[i]!=0) {
fuck=1;
}
}
if(fuck==0) {
return String.valueOf(1);
}
for (int i = 0 ; i < a.length - 2; i++) {
int head = i + 1, end = a.length - 1;
int n1=0;int n2=0;int n3=1;
while (true) {
if (head < end) {
if (a[i] + a[head] + a[end] > 0) {
end--;
endFlag=1;
headFlag=0;
} else if (a[i] + a[head] + a[end] < 0) {
head++;
headFlag=1;
endFlag=0;
} else if (a[i] + a[head] + a[end] == 0) {
if(a[i]!=n1||a[head]!=n2||a[end]!=n3) {
count++;
n1=a[i];
n2=a[head];
n3=a[end];
}
if(head==i+1) {
headFlag=1;
endFlag=0;
head++;
}
if(end==a.length-1) {
endFlag=1;
headFlag=0;
end--;
}
if(endFlag==1&&head < end--) {
end--;
}
else if(headFlag==1&&head++<end) {
head++;
}
}
} else {
break;
}
}
}
return String.valueOf(count);
}