题解 | #二叉搜索树的第k个结点#

二叉搜索树的第k个结点

http://www.nowcoder.com/practice/ef068f602dde4d28aab2b210e859150a

由于是二叉搜索树,所以一个结点左边的数都比该结点小,右边的数都比该结点大。 所以可以通过递归,按照中序遍历的顺序,先找到最左边的结点,该结点即为最小结点。然后开始计数,遍历到的第k个结点即为第k小结点。
这样做仅递归遍历了一次链表,大部分情况是不用遍历整个链表的,所以时间复杂度小于等于O(n)。递归过程中需要维护一个栈,空间复杂度等于O(logn)。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    bool start = false;
    int num = 0;
    TreeNode* result;
    TreeNode* KthNode(TreeNode* pRoot, int k) {
        if(!pRoot || !k){
            return nullptr;
        }
        num = k;
        find(pRoot);
        return result;
    }
    void find(TreeNode *p){
        if(!p->left){
            start = true;
        }
        if(p->left){
            find(p->left);
        }
        if(start){
            num--;
            if(num == 0){
                result = p;
            }
        }
        if(p->right){
            find(p->right);
        }
    }
};
全部评论

相关推荐

03-03 23:12
已编辑
北京邮电大学 Java
书海为家:我来给一点点小建议,因为毕竟还在学校不像工作几年的老鸟有丰富的项目经验,面试官在面试在校生的时候更关注咱们同学的做事逻辑和思路,所以最好在简历中描述下自己做过项目的完整过程,比如需求怎么来的,你对需求的解读,你想到的解决办法,遇到困难如何找人求助,最终项目做成了什么程度,你从中收获了哪些技能,你有什么感悟。
你的简历改到第几版了
点赞 评论 收藏
分享
卡卡罗特ovo:说起云智我就来气,约好了一面,结果面试官没来,ssob上问hr也未读,我还是专门请了半天假在家面试,恶心死了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务