ClassInstance

java List间计算并集、差集、交集以及去重

原创 2021-05-14 09:39:42.0 阅读(3723)次

Java中如何计算两个List的并集、差集、交集？
并集和交集我们都很清楚，但其中这里说的差集是什么意思呢？指的是集合listA与listB差集，准确来说是用listB去删除listA中存在listB中的元素。

Java集合中针对这三个场景提供了三个方法。
- 并集使用list.addAll()方法；
- 差集使用list.removeAll()方法；
- 交集使用list.retainAll()方法；

其中removeAll和retainAll都会根据对象的equals方法来比对对象是否相等，要正确的使用这两个方法需要注意对象的类是否要重写equals和hashCode方法。

下面分享一下如何实现两个List的并集、差集、交集，包括List的去重

具体代码如下：
ListJoinAndDiffAndInter.java
```java
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

import com.alibaba.fastjson.JSON;
import com.mystudy.model.UserInfo;

/**
 * 两个List的并集、差集、交集计算，去重方法
 * 差集可以用来去重 对象的比对需要重写equals和hashCode方法
 * @author Administrator
 *
 */
public class ListJoinAndDiffAndInter {

/**
	 * 求两个列表的并集
	 */
	public static void getListJoin() {

List<UserInfo> listA = new ArrayList<UserInfo>();
		List<UserInfo> listB = new ArrayList<UserInfo>();

UserInfo a = new UserInfo(1, "a", null);
		UserInfo b = new UserInfo(2, "b", null);
		UserInfo c = new UserInfo(3, "c", null);
		UserInfo d = new UserInfo(4, "d", null);

listA.add(a);
		listA.add(b);
		listA.add(c);

listB.add(b);
		listB.add(c);
		listB.add(d);

listA.addAll(listB);
		//去重
		List list = removeDupli(listA);

System.out.println(JSON.toJSONString(list));
	}

/**
	 * 求两个列表的差集，差集结果以listA为标准，可用来两个集合的去重
	 */
	public static void getListDiff() {

List<UserInfo> listA = new ArrayList<UserInfo>();
		List<UserInfo> listB = new ArrayList<UserInfo>();

UserInfo a = new UserInfo(1, "a", null);
		UserInfo b = new UserInfo(2, "b", null);
		UserInfo c = new UserInfo(3, "c", null);
		UserInfo d = new UserInfo(4, "d", null);

listA.add(a);
		listA.add(b);
		listA.add(c);

listB.add(b);
		listB.add(c);
		listB.add(d);

// removeAll需要UserInfo类重写equals和hashCode方法
		listA.removeAll(listB);
		System.out.println(JSON.toJSONString(listA));
	}

/**
	 * 求两个列表的交集
	 */
	public static void getListInter() {

List<UserInfo> listA = new ArrayList<UserInfo>();
		List<UserInfo> listB = new ArrayList<UserInfo>();

UserInfo a = new UserInfo(1, "a", null);
		UserInfo b = new UserInfo(2, "b", null);
		UserInfo c = new UserInfo(3, "c", null);
		UserInfo d = new UserInfo(4, "d", null);

listA.add(a);
		listA.add(b);
		listA.add(c);

listB.add(b);
		listB.add(c);
		listB.add(d);

// retainAll需要UserInfo类重写equals和hashCode方法
		listA.retainAll(listB);
		System.out.println(JSON.toJSONString(listA));
	}

/**
	 * 去重
	 * @param listA
	 * @return
	 */
	public static List removeDupli(List<UserInfo> listA) {
		// 去重,distinct需要UserInfo类重写equals和hashCode方法
		List<UserInfo> list = listA.stream().distinct().collect(Collectors.toList());
		return list;
	}

public static void main(String[] args) {

// 并集
		getListJoin();
		// 差集
		getListDiff();
		// 交集
		getListInter();

}

}

```

UserInfo.java
```java
@Data
public class UserInfo {

private Integer id;
	
	private String userName;
	
	private Integer deptId;
	
	public UserInfo() {
	}

public UserInfo(Integer id, String userName, Integer deptId) {
		super();
		this.id = id;
		this.userName = userName;
		this.deptId = deptId;
	}
	
    @Override
    public boolean equals(Object obj) {
        if (obj instanceof UserInfo) {
        	UserInfo tmp = (UserInfo) obj;
            if (this.getId().equals(tmp.getId())) {
                return true;
            }
        }
        return false;
    }

@Override
    public int hashCode() {
        return this.getId().hashCode();
    }
}

```

执行结果如下：
```
并集:[{"id":1,"userName":"a"},{"id":2,"userName":"b"},{"id":3,"userName":"c"},{"id":4,"userName":"d"}]
差集:[{"id":1,"userName":"a"}]
交集:[{"id":2,"userName":"b"},{"id":3,"userName":"c"}]
```

java java基础

上一篇：LinkedBlockingQueue实现定长队列自动出队与读取队列

下一篇：java Md5工具类获取字符串的Md5值