过滤HashMap的条目并对值进行迭代

Question

我使用了HashMap来实现图形

HashMap<E,Node<E>> graph=new HashMap<E,Node<E>>()

现在我想过滤掉它的头节点（Node类有一个方法，可以显示这个节点是头节点还是尾节点）

public Iterator<Node<E>> heads() {
    HashMap<E, Node<E>> heads=new HashMap<E, Node<E>>();
    for(Entry<E, Node<E>> e:graph.entrySet()){
        if(e.getValue().isHead())
            heads.put(e.getKey(), e.getValue());
    }
    return heads.values().iterator();
}

我想知道这是一种好的实施方式，还是有更好的适当方式？

Answer 1

不知道你在寻找"Better"的哪个方面。不管怎样，你的方式是有效的。

我更喜欢另一种方式，使用Guava的Iterables。

它看起来像：

Iterable<Node<E>> heads 
     = Iterables.filter(graph.values(), 
                        new Predicate<Node<E>> {
                            @Override
                            public boolean apply(Node<E> node) {
                                return node.isHead();
                            }
                       );

---

编辑：如果您使用的是Java8，那么您已经找到了类似的功能，而没有第三方库。多亏了Lambda，语法更加简单。不管怎样，这个想法是一样的。

Answer 2

与您发布的算法相比，一个直接的改进是将地图的头部元素收集到一个列表中，而不是重建其中只有头部元素的HashMap：

public Iterator<Node<E>> heads() {
    List<Node<E>> heads = new ArrayList<>();
    for (Node<E> e : graph.values()) {
        if (e.isHead())
            heads.add(e);
    }
    return heads.iterator();
}

Answer 3

Java 8版本：

Iterator<Node> s = graph.values().stream().filter(n -> n.isHead()).iterator();