int数组交并差集

最新推荐文章于 2021-05-21 03:18:16 发布

yannischen

最新推荐文章于 2021-05-21 03:18:16 发布

阅读量137

点赞数

分类专栏： java 文章标签： java

java 专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一个Java方法，该方法接收两个整数数组和一个字符串作为参数，通过使用TreeSet进行并集、交集和对称差集的操作。文章展示了如何通过流(Stream)API将数组转换为集合，并根据不同操作类型返回相应的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

public int[] operate(int[] a, int[] b, String operation){
		
		Set<Integer> aSet = new TreeSet<>(Arrays.stream(a).boxed().collect(Collectors.toList()));
		Set<Integer> bSet = new TreeSet<>(Arrays.stream(b).boxed().collect(Collectors.toList()));
		Set<Integer> resultSet = new TreeSet<>();
		switch (operation) {
			case "UNION":			//并集
				aSet.addAll(bSet);
				resultSet = aSet;
				break;
			case "INTERSECTION":	//交集
				resultSet.addAll(aSet);
				resultSet.retainAll(bSet);
				break;
			case "SYMMETRIC DIFFERENCE":	//差集
				resultSet.addAll(aSet);
				resultSet.removeAll(bSet);
				Set<Integer> removeSet = new TreeSet<>();
				removeSet.addAll(bSet);
				removeSet.removeAll(aSet);
				resultSet.addAll(removeSet);
				break;
	
			default:
				break;
		}
		int[] resultInt = new int[resultSet.size()];
		Iterator<Integer> it = resultSet.iterator();
		int i = 0;
		while (it.hasNext()) {
			resultInt[i] = it.next();
			i++;
		}
		return resultInt;
	}