前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Leetcode 230. Kth Smallest Element in a BST

Leetcode 230. Kth Smallest Element in a BST

作者头像
xindoo
发布2021-01-22 11:34:04
3460
发布2021-01-22 11:34:04
举报
文章被收录于专栏:XINDOO的专栏XINDOO的专栏

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it. 求一个二叉搜索树的第k小值。

题目链接:https://leetcode.com/problems/kth-smallest-element-in-a-bst/

  先来看下二叉搜索树的性质,对于任意一个非叶子节点,它的左子树中所有的值必定小于这个节点的val,右子树中所有的值必定大于或等于当前节点的val。 这条性质就保证了如果我们对二叉搜索树做中序遍历,中序遍历的结果肯定是有序的。对于此题而言,我们只需要拿到中序遍历结果,然后返回第k个即可,时间复杂度是O(n)。   但是,其实是有更优的解决方案的,我们只需要第k的数,并不需要遍历整个BST。这个时候,我们只需要把中序遍历的递归代码稍作改造即可,加一个全局遍历count来计数当前遍历到第几个数。因为上面所说的性质,中序遍历是有序的,所以只需要在计数器到k时返回当前节点的val,这就是我们需要的结果,时间复杂度O(k)。

代码语言:javascript
复制
/**
* Definition for a binary tree node.
* public class TreeNode {
*     int val;
*     TreeNode left;
*     TreeNode right;
*     TreeNode(int x) { val = x; }
* }
*/
public class Solution {
    private int count = 0;
    public int kthSmallest(TreeNode root, int k) {
        if (root == null)
            return 0;
        int ret=kthSmallest(root.left, k);
        if (0 != ret)
            return ret;
        if(++count == k) 
            return root.val;
        return kthSmallest(root.right, k);
   }
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2016-06-25,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档