最近做的一个项目,获取检查两个request中的选项是否一样,所以碰到了校验两个List是否相等的问题,所以在此yy了一下。

我们看看如何比较两个数组相等。数组是一个连续的内存空间,所以一般来说,两个数组相等,就是意味着他们有相同的长度,相同的元素,以及相同的顺序。我们看看JDK的Arrays.equals()实现就一目了然了。

public static boolean equals(int[] a, int[] a2) {
        if (a==a2) return true;
        if (a==null || a2==null) return false;
        int length = a.length;
        if (a2.length != length) return false;
        for (int i=0; i<length; i++)
            if (a[i] != a2[i]) return false;
        return true;
    }

         大致分成下面4个步骤:

1. 检查是否指向同一个地址(不同引用);

2. 非空检查;

3. 长度检查;

4. 顺序和对应的元素相等检查。(依次比较,只要一个不等就返回false)        

而对于List来说,没有固定的顺序(ArrayList底层是数组,可以用数组的比较方法;但是其他的List,如LinkedList,就不一样了。),或者说顺序对List来说不是一个重要的属性。所以可以考虑其他方式来检查两个List是不是包含相同的元素。

如JDK的Collection.containsAll(),只是是对每个元素进行contains()操作。但是List允许重复的元素,所以这里无法比较重复元素的数量:

public boolean containsAll(Collection<?> c) {
        for (Object e : c)
            if (!contains(e))
                return false;
        return true;
    }

 那么,如何对无序的元素进行equal比较呢?

比较简便直观的方法,就是分别对两个list排序,然后一个一个比较,只要不相等,则返回false,如果全部都相等,则返回true。

if (a.size() !=b.size())
    return false;
Collections.sort(a);
Collections.sort(b);
for (int i = 0; i <a.size(); i++) {
    if(!a.get(i).equals(b.get(i)))
       return false;
}
return true;

 但是这样两次排序,效率还是比较低的。所以可以通过Map来实现:

先存入HashMap,key为元素,value为出现的次数,然后再逐个元素比较出现的次数,这样能确保元素都包含,而且出现次数相同。

如下代码所示,具体可以参考org.apache.commons.collections.CollectionUtils.isEqualCollection() 。(CollectionUtils中大量使用了getCardinalityMap()的特性)

private static final Integer INTEGER_ONE = 1;
public static boolean isEqualCollection(Collection a, Collection b){
    if (a.size() !=b.size()) {  // size是最简单的相等条件
       return false;
    }
    Map mapa = getCardinalityMap(a); 
    Map mapb = getCardinalityMap(b);
   
    // 转换map后,能去掉重复的,这时候size就是非重复项,也是先决条件
    if (mapa.size() !=mapb.size()) {  
       return false;
    }
    Iterator it =mapa.keySet().iterator();
    while (it.hasNext()) {
       Object obj = it.next();
       // 查询同一个obj,首先两边都要有,而且还要校验重复个数,就是map.value
       if (getFreq(obj,mapa) != getFreq(obj, mapb)) {
           return false;
       }
    }
    return true;
}
/**
 * 以obj为key,可以防止重复,如果重复就value++
 * 这样实际上记录了元素以及出现的次数
 */
public static Map getCardinalityMap(Collection coll) {
    Map count = new HashMap();
    for (Iterator it =coll.iterator(); it.hasNext();) {
       Object obj =it.next();
       Integer c =(Integer) count.get(obj);
       if (c == null)   
           count.put(obj, INTEGER_ONE);
       else {
           count.put(obj, newInteger(c.intValue() + 1));
       }
    }
    return count;
}
private static final int getFreq(Objectobj, Map freqMap) {
    Integer count =(Integer) freqMap.get(obj);
    if (count != null) {
       return count.intValue();
    }
    return 0;
}