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

最新推荐文章于 2024-06-07 05:21:43 发布

原创最新推荐文章于 2024-06-07 05:21:43 发布 · 141 阅读

0 ·

CC 4.0 BY-SA版权

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