// uses HashMap by default static class MultiSet { Map map = new HashMap; public MultiSet(bool useTreeMap) { if (useTreeMap) map = new TreeMap; } public MultiSet() { } public MultiSet(Collection c) { addAll(c); } public void add(A key) { add(key, 1); } public void addAll(Collection c) { if (c != null) for (A a : c) add(a); } void addAll(MultiSet ms) { for (A a : ms.keySet()) add(a, ms.get(a)); } public void add(A key, int count) { if (map.containsKey(key)) map.put(key, map.get(key)+count); else map.put(key, count); } public int get(A key) { return key != null && map.containsKey(key) ? map.get(key) : 0; } public bool contains(A key) { ret map.containsKey(key); } public void remove(A key) { Integer i = map.get(key); if (i != null && i > 1) map.put(key, i - 1); else map.remove(key); } public List topTen() { ret getTopTen(); } public List getTopTen() { ret getTopTen(10); } public List getTopTen(int maxSize) { List list = getSortedListDescending(); return list.size() > maxSize ? list.subList(0, maxSize) : list; } List highestFirst() { ret getSortedListDescending(); } List lowestFirst() { ret reversedList(getSortedListDescending()); } public List getSortedListDescending() { List list = new ArrayList(map.keySet()); Collections.sort(list, new Comparator() { public int compare(A a, A b) { return map.get(b).compareTo(map.get(a)); } }); return list; } public int getNumberOfUniqueElements() { return map.size(); } int uniqueSize() { ret map.size(); } public Set asSet() { return map.keySet(); } Set keySet() { // synonym for idiots return map.keySet(); } public A getMostPopularEntry() { int max = 0; A a = null; for (Map.Entry entry : map.entrySet()) { if (entry.getValue() > max) { max = entry.getValue(); a = entry.getKey(); } } return a; } public void removeAll(A key) { map.remove(key); } public int size() { int size = 0; for (int i : map.values()) size += i; return size; } public MultiSet mergeWith(MultiSet set) { MultiSet result = new MultiSet(); for (A a : set.asSet()) { result.add(a, set.get(a)); } return result; } public boolean isEmpty() { return map.isEmpty(); } public String toString() { return str(map); } public void clear() { map.clear(); } Map asMap() { ret cloneMap(map); } }