码迷,mamicode.com
首页 > 其他好文 > 详细

Tree inorder traversal Iterator

时间:2014-11-24 07:35:51      阅读:143      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   sp   on   div   log   bs   代码   

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;
    }
};

 

Tree inorder traversal Iterator

标签:style   blog   color   sp   on   div   log   bs   代码   

原文地址:http://www.cnblogs.com/jobfindingnotes/p/4117873.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!