在我们的日常开发的过程中,经常会遇到一些问题需要我们进行一些集合的运算,这里我们对集合中最为常用的交集、并集、差集这几种运算进行了一个简单的总结:
交集的实现
/**
* 计算交集
* @param s1
* @param s2
* @return
*/
public static Set<Object> intersection(Set<Object> s1,Set<Object> s2){
Set<Object> result = new HashSet<Object>();
result.clear();
result.addAll(s1);
result.retainAll(s2);
return result;
}
测试及结果
public static void main(String[] args) {
Set<Object> s1 = new HashSet<Object>();
s1.add(1);
s1.add(2);
s1.add(3);
Set<Object> s2 = new HashSet<Object>();
s2.add(2);
s2.add(3);
s2.add(4);
//集合中的交集测试
Set<Object> result = TestController.intersection(s1,s2);
if (!result.isEmpty()) {
for (Object o : result) {
System.out.println(o.toString());
}
}
}
运行结果:
Java8 Stream 中交集的实现
public static void main(String[] args) {
List<String> list1 = Lists.newArrayList("a", "b", "c");
List<String> list2 = Lists.newArrayList("a", "b", "d", "e");
//交集
List<String> list3 = list1.stream().filter(list2::contains).collect(Collectors.toList());
System.out.println(list3);
}
并集的实现
<pre name="code" class="java">/**
* 计算并集
* @param s1
* @param s2
* @return
*/
public static Set<Object> union(Set<Object> s1,Set<Object> s2){
Set<Object> result = new HashSet<Object>();
result.clear();
result.addAll(s1);
result.addAll(s2);
return result;
}
测试及结果
public static void main(String[] args) {
Set<Object> s1 = new HashSet<Object>();
s1.add(1);
s1.add(2);
s1.add(3);
Set<Object> s2 = new HashSet<Object>();
s2.add(2);
s2.add(3);
s2.add(4);
//集合中的并集测试
Set<Object> result = TestController.union(s1,s2);
if (!result.isEmpty()) {
for (Object o : result) {
System.out.println(o.toString());
}
}
}
运行结果:
差集的实现
/**
* 计算差集
* @param s1
* @param s2
* @return
*/
public static Set<Object> difference(Set<Object> s1,Set<Object> s2){
Set<Object> result = new HashSet<Object>();
result.clear();
result.addAll(s1);
result.removeAll(s2);
return result;
}
测试及结果
public static void main(String[] args) {
Set<Object> s1 = new HashSet<Object>();
s1.add(1);
s1.add(2);
s1.add(3);
Set<Object> s2 = new HashSet<Object>();
s2.add(2);
s2.add(3);
s2.add(4);
//集合中的并集测试
Set<Object> result = TestController.difference(s1,s2);
if (!result.isEmpty()) {
for (Object o : result) {
System.out.println(o.toString());
}
}
}
运行结果
Java8 Stream 差集的实现
public static void main(String[] args) {
List<String> list1 = Lists.newArrayList("a", "b", "c");
List<String> list2 = Lists.newArrayList("a", "b", "d", "e");
//差集: list1 - list2
List<String> list4 = list1.stream().filter(i -> !list2.contains(i)).collect(Collectors.toList());
System.out.println(list4);
//差集: list2 - list1
List<String> list5 = list2.stream().filter(i -> !list1.contains(i)).collect(Collectors.toList());
System.out.println(list5);
}