将 TreeMap 修剪为 n 个条目

Question

我有一个已排序的 TreeMap，我想删除除前 10 个元素之外的所有元素。有什么方法可以做到这一点？我考虑过在地图大小大于 10 时从地图末尾删除元素，但我不知道该怎么做。我可以转换为列表然后再转换回来吗？

谢谢

Answer 1

HashMap 没有开头或结尾——它们是无序的。迭代项目的顺序与插入项目的顺序完全无关。如果您可以改用 LinkedHashMap，那么该类实际上会保留项目插入的顺序。然后，您可以简单地迭代 entrySet()，在迭代前 10 个项目后，对每个项目调用迭代器上的 remove()。

Answer 2

这是一个想法。也许制作一个大小限制为 10 个元素的地图类，并以这种方式复制/构造有限的地图段：

package main;

import java.util.LinkedHashMap;
import java.util.Map;

import com.google.common.collect.Maps;

public class HashMap10<K, V> extends LinkedHashMap<K, V> {

    private static final long serialVersionUID = -4943383947326287590L;

    static Map<Integer, String> x = Maps.newHashMap();

    public HashMap10() {
        super();
    }

    public HashMap10(int initialCapacity, float loadFactor, boolean accessOrder) {
        super(initialCapacity, loadFactor, accessOrder);
    }

    public HashMap10(int initialCapacity, float loadFactor) {
        super(initialCapacity, loadFactor);
    }

    public HashMap10(int initialCapacity) {
        super(initialCapacity);
    }

    public HashMap10(Map<? extends K, ? extends V> m) {
        putAll(m);
    }

    @Override
    public V put(K key, V value) {
        if (this.size() == 10) {
            return null;
        }
        return super.put(key, value);
    }

    @Override
    public void putAll(Map<? extends K, ? extends V> m) {
        for (java.util.Map.Entry<? extends K, ? extends V> x : m.entrySet()) {
            put(x.getKey(), x.getValue());
        }
    }

}

通过向有限大小的地图添加 100 个元素来测试它，一次通过多次 put() 调用，一次使用构造函数：

package main;

import java.util.HashMap;
import java.util.Map.Entry;

public class TestIt {

    /**
     * @param args
     */
    public static void main(String[] args) {
        HashMap10<Integer, String> map10 = new HashMap10<Integer, String>();

        for (int i = 1; i < 99; i++) {
            map10.put(i, Integer.toString(i + 100));
        }

        for (Entry<Integer, String> x : map10.entrySet()) {
            System.out.println(x.getKey() + "->" + x.getValue());
        }
        System.out.println("");
        System.out.println("");

        HashMap<Integer, String> mapUnlimited = new HashMap<Integer, String>();

        for (int i = 1; i < 99; i++) {
            mapUnlimited.put(i, Integer.toString(i + 200));
        }

        HashMap10<Integer, String> anotherMap10 = new HashMap10<Integer, String>(
                mapUnlimited);
        for (Entry<Integer, String> x : anotherMap10.entrySet()) {
            System.out.println(x.getKey() + "->" + x.getValue());
        }

    }

}

转储映射，无论您一次添加一个元素，还是使用构造函数构建映射，您都只有 10 个元素的映射：

1->101
2->102
3->103
4->104
5->105
6->106
7->107
8->108
9->109
10->110

1->201
2->202
3->203
4->204
5->205
6->206
7->207
8->208
9->209
10->210

我没有在防御性编程、检查空值之类的事情上投入任何时间。这里的想法是我不知道也不关心你的原始地图是什么。链接地图。树图。常规地图。无论它返回的元素的顺序如何，都定义了“前 10 个元素”的含义，而我的地图类将仅存储前 10 个元素，并忽略之后的任何元素。现在您已经有了前 10 个，无论这对您的地图意味着什么。