在BST中使用单个孩子计算节点

时间:2015-05-30 19:04:52

标签: c++ binary-search-tree tree-traversal

我有这个问题:实现函数int countSingle(),它使用广度优先遍历(BFT)来计算树中有多少个节点有一个子节点。

所以下面的代码就是我想要解决的问题,有没有另外一种方法可以做到这一点或更高效,我错过了?

template <class T>
        int BST<T>::countSingle()
        {
            QueueDLL<BSTNode<T>*> queue;
            BSTNode<T>* node = root;

            // If the tree is empty, there will be no nodes to traverse.
            if(node == NULL) return -1;

            // Initially, put the root in the queue
            queue.Enqueue(node);
            int counter = 0 ;

            while(queue.IsEmpty() == false)
            {
                // Take a node out of the queue, process it and then insert
                // its children into the queue.
                node = queue.Dequeue();

                // if the node has one only one child wether its the left or the right , increase counter
                if((node->left == NULL && node->right != NULL) || (node->right== NULL && node->left!=NULL))
                    counter++;

                if(node->left != NULL)
                    queue.Enqueue(node->left);
                if(node->right != NULL)
                    queue.Enqueue(node->right);
            }
            return counter;
        }

1 个答案:

答案 0 :(得分:0)

这似乎很好。你无法真正提高你所做的复杂性,因为操作的数量是线性的。

一些实施要点:

  • 计数是一种非修改操作,因此您可能需要考虑const某些内容。

  • 你基本上已经实现了类似于编译器在递归中所做的事情。如果这是意图 - 很棒(并且有一些性能点对你有利)。仅供参考,您可以使用极少数LOC的递归来完成此操作。