小米OJ上分日志——（十五）和为零的三元组

序号：#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 ***=0;
		for(int i=0;i<a.length;i++) {
   
			if(a[i]!=0) {
   
				***=1;
			}
		}
		if(***==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);
	}