分类->蓝桥杯JAVA练习->简单题
题目 1606: 蓝桥杯算法训练VIP-集合运算
解题思路:
我采用了ArrayList,就不需要自己写交并差查找的算法了,用的是Java8 Stream
具体实现参考了Qblue666的文章Java中交集、并集、差集、补集、去重的实现_java 差集_Qblue666的博客-CSDN博客
可以把这个改成此类问题的模板使用
代码参考:
import java.util.*;
import java.util.stream.Collectors;
public class Main
{
public static void main(String[] args)
{
Scanner sc=new Scanner(System.in);
//输入第一个集合
int n1=sc.nextInt();
ArrayList<Integer> a = new ArrayList<Integer>(); //不是int,而是Integer
for(int i=0;i<n1;i++)
{
int x=sc.nextInt();
a.add(x);
}
//输入第二个集合
int n2=sc.nextInt();
ArrayList<Integer> b = new ArrayList<Integer>();
for(int i=0;i<n2;i++)
{
int x=sc.nextInt();
b.add(x);
}
//a*b交集
List<Integer> jiao = a.stream().filter(b::contains).collect(Collectors.toList());
Collections.sort(jiao);
System.out.println(Arrays.toString(jiao.toArray()).replace("[", "").replace("]", "").replace(",", ""));
//a+b并集
ArrayList<Integer> bing = new ArrayList<Integer>();
bing.addAll(a);
bing.removeAll(b);//先从本集合中删除另一个集合中相同的元素
bing.addAll(b);//再把另外一个集合中所有元素,添加到本集合中
Collections.sort(bing);
System.out.println(Arrays.toString(bing.toArray()).replace("[", "").replace("]", "").replace(",", ""));
//a-b的差集
List<Integer> cha = a.stream().filter(i -> !b.contains(i)).collect(Collectors.toList());
Collections.sort(cha);
System.out.println(Arrays.toString(cha.toArray()).replace("[", "").replace("]", "").replace(",", ""));
}
}
总结:
1、改写成集合运算的模板
2、控制列表输出格式的时候,简洁的做法是使用Arrays.toString把列表输出转化为字符串,然后使用replace()把不要的元素替换掉