扩展时迭代自己类型的Java类

Question

我想创建一个自定义的树数据结构，只有节点，我可以迭代他们。然后，我可以扩展这个类，并有非常基本的树

class Node{

    Node parent;
    ArrayList<Node> children;

    public static void main(String[]args){
        Node root = new Node();
        for(Node child : root){
            //do something
        }
    }

    public Iterator<Node> iterator(){
        // basic tree traversal iterator
    }  
}

我已经开始工作了，但是当我试图扩展Node类时，问题就来了。对于扩展类，继承的迭代器方法仍然返回节点迭代器，这意味着我每次都要强制转换。下面是我遇到的问题的一个基本示例。让我们建立一个包含整数的树：

class IntegerNode extends Node{

    int value;

    public static void main(String[]args){

        IntegerNode root = new IntegerNode();
        int total = 0;

        for(IntegerNode child : root){  /* Compiler error, says that the
            iterator returns Iterator<Node> and not Iterator<IntegerNode>*/
            total+=child.value;
        }

        System.out.println(total);
    }

}

是否有一种简单的方法可以解决这个问题，而不需要将iterator（）方法从Node类复制到IntegerNode类中？

Answer 1

我认为以下将工作（未经测试，所以不是100%肯定）：

class Node<T extends Node<T>> {
    public Iterator<T> iterator(){
        // basic tree traversal iterator
    }  
}

class IntegerNode extends Node<IntegerNode> {
    public static void main(String[]args) {
        IntegerNode root = new IntegerNode();
        int total = 0;
        for(IntegerNode child : root){  
            total += child.value;
        }

        System.out.println(total);
    }
}

这基本上是准标准可继承构建器模式的扩展。