Tree inorder traversal Iterator

二叉树中序遍历迭代器

最新推荐文章于 2022-02-26 19:51:19 发布

weixin_30247307

最新推荐文章于 2022-02-26 19:51:19 发布

阅读量53

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/jobfindingnotes/p/4117873.html

本文介绍了一个用于二叉树中序遍历的迭代器实现方法。通过使用栈来辅助迭代过程，该迭代器能够有效地进行二叉树节点的中序访问。文章中的代码示例展示了如何定义一个TreeNode类，并实现hasNext和next成员函数。

TreeNode a;
while (a.hasNext())
visit(a.next());
}"
问：给TreeNode写Iterator，使得以上代码可以in order traversal

class TreeNode
{
TreeNode *root;
public:
    bool hasNext()
    {
        return root!=NULL || !stack.empty();
    }
    TreeNode * next()
    {
        TreeNode * ret = NULL;
        while(hashNext())
        {
            if(root)
            {
                stack.push(root);
                root = root->left;
            }
            else
            {
                root = stack.top();
                stack.pop();
                ret = root;
                root = root->right;
                return ret;
            }
        }
        return NULL;
    }
};