【leetcode】173.（Medium）Binary Search Tree Iterator

AXIMI

于 2019-04-05 13:49:35 发布

阅读量162

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/AXIMI/article/details/89042804

leetcode 专栏收录该内容

257 篇文章

订阅专栏

本文介绍了一种使用Morris遍历算法实现二叉搜索树(BST)迭代器的方法。通过一次中序遍历即可获取BST的所有节点值，并以此为基础实现了next()和hasNext()方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解题思路：

morris遍历先中序把整棵树遍历一遍

提交代码：

class BSTIterator {
	private List<Integer> nums=new ArrayList<Integer>();
	private int p=0;
	
    public BSTIterator(TreeNode root) {
        TreeNode cur=root,tmp=root;
        
        while(cur!=null) {
        	if(cur.left!=null) {
        		tmp=cur.left;
        		while(tmp.right!=null&&tmp.right!=cur)
        			tmp=tmp.right;
        
        			if(tmp.right==null) {
        				tmp.right=cur;
        				cur=cur.left;
        			}else {
        				nums.add(cur.val);
        				cur=cur.right;
        			}
        
        	}else {
        		nums.add(cur.val);
        		cur=cur.right;
        	}
        }
    }
    
    /** @return the next smallest number */
    public int next() {
        return nums.get(p++);
    }
    
    /** @return whether we have a next smallest number */
    public boolean hasNext() {
        if(p<nums.size())	return true;
        return false;
    }
}