1. 程式人生 > >和為零的三元組

和為零的三元組

描述
給出一個整數陣列, 陣列中是否存在任意 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());

	}
}