和为零的三元组

描述
给出一个整数数组, 数组中是否存在任意 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());

	}
}

猜你喜欢

转载自blog.csdn.net/adityasaber/article/details/82945886
今日推荐