读书人

Java里面怎么求两个集合的交集

发布时间: 2012-12-21 12:03:49 作者: rapoo

Java里面如何求两个集合的交集

? ? ?在Python里,或许我们没有这个烦恼,因为python里已经为我们提供了intersection这样的方法。

? ? ?但是在Java里,就需要我们动一番脑筋了。这里浓重推荐下apache的CollectionUtils工具类。

?

?

? ? ?方法签名如下所示:

? ? ?org.apache.commons.collections.intersection(final Collection a, final Collection b)

?

? ? ?那么这个方法是怎么实现的呢?这里以list为例

?

? ? ?public class TestIntersection {

    private static final Integer ONE = new Integer(1);    public static void main(String[] args) {        // a集合[a,a,b,b,b,c]        List<String> a = Arrays.asList("a", "a", "b", "b", "b", "c");        // b集合[a,b,b,c,c]        List<String> b = Arrays.asList("a", "b", "b", "c", "c");        Map mapa = mapForCollection(a);        Map mapb = mapForCollection(b);        // 将两个集合里不重复的元素加进来,然后会依次遍历元素的出现次数        Set set = new HashSet(a);        set.addAll(b);        List<String> result = new ArrayList<String>();        for (Object obj : set) {            for (int i = 0, m = Math.min(getCountsFromMap(obj, mapa), getCountsFromMap(obj, mapb)); i < m; i++) {                result.add((String) obj);            }        }        // 看下期待的结果是不是[a,b,b,c]        System.out.println(result);    }    /**     * 循环遍历集合,并对每一个元素出现的次数计数<br/>     * 最终返回类似于{A:1,B:3,C:3}这样的map     *      * @param a     * @return     */    private static Map mapForCollection(Collection a) {        Map map = new HashMap();        Iterator it = a.iterator();        while (it.hasNext()) {            Object obj = it.next();            Integer count = (Integer) map.get(obj);            if (count == null) {                // 表明该元素第一次出现                map.put(obj, ONE);            } else {                map.put(obj, new Integer(count.intValue() + 1));            }        }        return map;    }    private static int getCountsFromMap(Object obj, Map map) {        Integer count = (Integer) map.get(obj);        return count != null ? count.intValue() : 0;    }}
?

? ? 可以看到,先对两个不同的集合进行元素标记,并记下各自每个元素出现的次数,然后提取出两个集合中不重复的元素,

取两者中元素出现次数最少的数值,进行循环添加

?

读书人网 >编程

热点推荐