Question

I need to find the number of elements in a tree using an iterative algorithm, but I’m finding the code conceptually very difficult to write.

My approach is to start at the root node and visit the child nodes, then the children of these child nodes, and so on.

This is the code I’ve written which works for a small tree, but isn’t a real solution because I’d need to add an additional block for each level of depth:

// Start the counter at 1 because the root node counts
int size = 1;

for(ITree child1 : root) {
    size++;
    for(ITree child2 : child1) {
        size++;
        for(ITree child3 : child2) {
            size++;
            for(ITree child4 : child3) {
                size++;
                for(ITree child5 : child4) {
                    size++;
                }
            }
        }
    }
}
return size;

Answer 1

Conceptually, keep a stack (LinkedList, etc.). For each child (now, your child loops), add to the stack. Continue looping through the stack until it is finally empty.

This isn’t tested, but this should do exactly what you’re looking for. I’m just using java.io.File instead of your “ITree”, as it’s something I can compile against:

int sizeOfTree(File root){
    // Start the counter at 1 because the root node counts

    int size = 1;

    LinkedList<File> stack = new LinkedList<File>();
    stack.add(root);

    while(!stack.isEmpty()){
        File f = stack.remove();
        for(File child : f.listFiles()){
            size++;
            stack.add(child);
        }
    }

    return size;
}