从 Java 中的 TreeSet 中查找并返回一个元素

Question

我有以下节点类

Class Node {
    private int id;

    public int getId() {
        return this.id;
    }
}

然后用节点创建一个 TreeSet。接下来我想根据 id 匹配找到并返回一个 Node 对象。但是，每次 findNode() 函数都返回下一个节点而不是下一个节点。我知道这是因为调用了 iterator.next() 两次。如何只调用一次以检查 id 值并返回对象引用。我也尝试过创建一个临时对象引用，但结果还是一样。

Class NodeSet {
    Set<Node> set = new TreeSet<Node>();

    public Node findNode(int id) {  
        Iterator<Node> iterator = set.iterator();
        while(iterator.hasNext()) {
            if(iterator.next().getId() == id)               
                return iterator.next();
        }

        return null;                
    }
}

Answer 1

Class NodeSet {
    Set<Node> set = new TreeSet<Node>();

    public Node findNode(int id) {  
        Iterator<Node> iterator = set.iterator();
        while(iterator.hasNext()) {
            Node node = iterator.next();
            if(node.getId() == id)             
                return node;
        }

        return null;                
    }
}

Answer 2

问题发生在这里：

        while(iterator.hasNext()) {
            if(iterator.next().getId() == id)               
                return iterator.next();
        }

您在同一个循环中调用了两次 iterator.next，解释了“下一个到下一个”问题。

使局部变量仍能到达相同或更好的元素：如果您有 jdk >= 5，请使用 for 循环：

for(Node node: set) {
   if(node.getId() == id) 
     return node;
}

正如@JB Nizet 在上面的评论中所建议的那样，简单的Map 已经从本质上实现了您的代码逻辑，因此比TreeSet 和手动检查更适合。
更准确地说，TreeMapsorted on Nodes 将是相关的。 （因为听起来你需要订单方面）

Answer 3

编辑：这里提出的解决方案是对数（与公认的答案不同），因此当树集中的节点数量很大时会快得多。

TreeSet 类实现的Set 接口中没有get 方法。请记住，树集的元素之间存在完整的排序，三行 hack 如下：

Object search(TreeSet treeset, Object key) {
    Object ceil  = treeset.ceiling(key); // least elt >= key
    Object floor = treeset.floor(key);   // highest elt <= key
    return ceil == floor? ceil : null; 
}