和為零的三元組
阿新 • • 發佈:2018-11-09
描述
給出一個整數陣列, 陣列中是否存在任意 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()); } }