题解 | #二叉搜索树的第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);
        }
    }
};
全部评论

相关推荐

Pandaileee:校友加油我现在也只有一个保底太难了
点赞 评论 收藏
分享
过往烟沉:我说什么来着,java就业面就是广!
点赞 评论 收藏
分享
废铁汽车人:秋招真是牛鬼蛇神齐聚一堂
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务