【leetcode】230（Medium）Kth Smallest Element in a BST

AXIMI

于 2019-05-18 09:28:49 发布

阅读量125

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

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

leetcode 专栏收录该内容

257 篇文章

订阅专栏

博客介绍了求第k小的数的思路，即使用morris中序遍历，当遍历到第k个节点时，该节点的值就是所求的第k小的数。还提及了提交代码和运行结果，但未给出具体内容。

思路

使用morris中序遍历，遍历到第k个节点时就是第k小的数。

提交代码：

class Solution {
    public int kthSmallest(TreeNode root, int k) {
        TreeNode p1=root,p2=root;
        int cnt=0;
        while(p1!=null) {
        	if(p1.left!=null) {
        		p2=p1.left;
        		while(p2.right!=null&&p2.right!=p1)
        			p2=p2.right;
        		if(p2.right==null) {
        			p2.right=p1;
        			p1=p1.left;
        		}else {
        			p2.right=null;
        			cnt++;
        			if(cnt==k)	return p1.val;
        			p1=p1.right;
        		}
        				
        	}else {
        		cnt++;
        		if(cnt==k)	return p1.val;
        		p1=p1.right;
        	}
        }
        return root.val;
    }
}