【leetcode腾讯精选练习】9.二叉搜索树中第K小的元素

最新推荐文章于 2022-10-30 10:27:27 发布

AXIMI

最新推荐文章于 2022-10-30 10:27:27 发布

阅读量99

点赞数

本文链接：https://blog.csdn.net/AXIMI/article/details/103773769

版权

leetcode腾讯精选练习专栏收录该内容

17 篇文章 0 订阅

订阅专栏

使用morris遍历记录一下所有的节点

提交代码

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        TreeNode cur=root,next=root;
        ArrayList<Integer> values=new ArrayList<>();
        
        while(cur!=null) {
        	if(cur.left==null) {
        		values.add(cur.val);
        		cur=cur.right;
        	}else {
        		next=cur.left;
        		while(next.right!=null&&next.right!=cur)
        			next=next.right;
        		if(next.right==null) {
        			next.right=cur;
        			cur=cur.left;
        		}else {
        			values.add(cur.val);
        			cur=cur.right;
        			next.right=null;
        		}
        	}
        	
        }
        return values.get(k-1);
    }
}