以元素频率的顺序迭代Multiset的最简单方法是什么?

时间:2010-12-03 12:43:13

标签: java guava multiset

考虑这个打印出一些设备类型统计数据的例子。 (“DeviceType”是具有数十个值的枚举。)

Multiset<DeviceType> histogram = getDeviceStats();
for (DeviceType type : histogram.elementSet()) {
    System.out.println(type + ": " + histogram.count(type));
}

以频率顺序打印不同元素的最简单,最优雅的方式是什么?(最常见的类型是第一种)?

快速查看Multiset界面,没有现成的方法,也没有Guava的Multiset实现(HashMultisetTreeMultiset等)似乎自动保持频率排序的元素。

4 个答案:

答案 0 :(得分:39)

我刚刚将此功能添加到Guava中,请参阅here了解Javadoc。

修改:根据原始问题Multisets.copyHighestCountFirst()的使用示例:

Multiset<DeviceType> histogram = getDeviceStats();
for (DeviceType type : Multisets.copyHighestCountFirst(histogram).elementSet()) {
    System.out.println(type + ": " + histogram.count(type));
}

答案 1 :(得分:7)

这是一个返回List条目的方法,按频率排序( UPDATE :使用标志来切换升序/降序并使用Guava最喜欢的玩具:{{1} },如Effective Java中第3项所示:

Enum Singleton Pattern

测试代码:

private enum EntryComp implements Comparator<Multiset.Entry<?>>{
    DESCENDING{
        @Override
        public int compare(final Entry<?> a, final Entry<?> b){
            return Ints.compare(b.getCount(), a.getCount());
        }
    },
    ASCENDING{
        @Override
        public int compare(final Entry<?> a, final Entry<?> b){
            return Ints.compare(a.getCount(), b.getCount());
        }
    },
}

public static <E> List<Entry<E>> getEntriesSortedByFrequency(
    final Multiset<E> ms, final boolean ascending){
    final List<Entry<E>> entryList = Lists.newArrayList(ms.entrySet());
    Collections.sort(entryList, ascending
        ? EntryComp.ASCENDING
        : EntryComp.DESCENDING);
    return entryList;
}

<强>输出:

  

升序:
  一(1)
  二(2)
  三(3)
  四(4)
  降:
  四(4)
  三(3)
  二(2)
  一(1)

答案 2 :(得分:3)

使用ForwardingMultiSet实施

EntryComp 来自seanizer's answer

enum EntryComp implements Comparator<Multiset.Entry<?>> {
    DESCENDING {
        @Override
        public int compare(final Entry<?> a, final Entry<?> b) {
            return Ints.compare(b.getCount(), a.getCount());
        }
    },
    ASCENDING {
        @Override
        public int compare(final Entry<?> a, final Entry<?> b) {
            return Ints.compare(a.getCount(), b.getCount());
        }
    },
}

public class FreqSortMultiSet<E> extends ForwardingMultiset<E> {
    Multiset<E> delegate;
    EntryComp comp;

    public FreqSortMultiSet(Multiset<E> delegate, boolean ascending) {
        this.delegate = delegate;
        if (ascending)
            this.comp = EntryComp.ASCENDING;
        else
            this.comp = EntryComp.DESCENDING;
    }

    @Override
    protected Multiset<E> delegate() {
        return delegate;
    }

    @Override
    public Set<Entry<E>> entrySet() {
        TreeSet<Entry<E>> sortedEntrySet = new TreeSet<Entry<E>>(comp);
        sortedEntrySet.addAll(delegate.entrySet());
        return sortedEntrySet;
    }

    @Override
    public Set<E> elementSet() {
        Set<E> sortedEntrySet = new LinkedHashSet<E>();
        for (Entry<E> en : entrySet())
            sortedEntrySet.add(en.getElement());
        return sortedEntrySet;
    }

    public static <E> FreqSortMultiSet<E> create(boolean ascending) {
        return new FreqSortMultiSet<E>(HashMultiset.<E> create(), ascending);
    }

    /*
     * For Testing
     * public static void main(String[] args) {
        Multiset<String> s = FreqSortMultiSet.create(false);
        s.add("Hello");
        s.add("Hello");
        s.setCount("World", 3);
        s.setCount("Bye", 5);
        System.out.println(s.entrySet());
    }*/

}

答案 3 :(得分:2)

it is not yet implemented以来,我猜您可以使用key = type和value = count创建Map。然后对该地图进行排序 - 请参阅here