easy 剑指 Offer 二叉搜索树的第k大节点 中序遍历 栈 递归

二叉搜索树(二叉查找树、二叉排序树)

在这里插入图片描述

二叉搜索树(二叉查找树、二叉排序树):

每棵子树头节点的值都比各自左子树上所有节点值要大,也都比各自右子树上所有节点值要小。二叉查找树的中序遍历序列一定是从小到大排列的。

平衡二叉查找树:

任何一个结点的左子树与右子树都是平衡二叉树,并且高度之差的绝对值不超过1


在这里插入图片描述


中序遍历 栈:

二叉查找树的中序遍历序列一定是从小到大排列的

c++


class Solution {
public:
    int kthLargest(TreeNode* root, int k) {
        vector<int> res;
        stack<TreeNode*> stk;  // 栈 放遍历到的左节点
        while(root!=nullptr || !stk.empty()){     // 节点为空且栈空,则已访问完最右节点
            while(root != nullptr){      // 找到最左节点
                stk.push(root);  // 将遍历到的左节点放到栈中
                root = root->left;  // 遍历下一个左节点
            }
            root = stk.top(); // 开始回溯,此时栈顶为最左节点(该根下面没有左节点),输出该根
            res.push_back(root->val);
            root = root->right;   // 访问该根右节点
            stk.pop();   // 访问完毕,出栈
        }

        return res[res.size()-k];
    }
};

python


class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
        if not root:
            return 0
        res= []
        stack = []
        while root or stack:     # 栈不为空就循环
            while root:
                stack.append(root)
                root = root.left
            root = stack[-1]
            res.append(root.val)   # 左
            root = root.right      # 有右节点,下一次while就会找该右节点 最左根;没有右节点root=null,无while循环
            stack.pop()            # 访问完毕,出栈

        return res[len(res)-k]

在这里插入图片描述


递归

c++


class Solution {
public:
    void inorder(TreeNode* root, vector<int>& res){  // 传入向量指针 &res
        if(root==NULL){  //  递归终止的条件为碰到空节点
            return;   
        }
        inorder(root->left,res);   // 左中(根)右
        res.push_back(root->val);
        inorder(root->right,res);
    }


    int kthLargest(TreeNode* root, int k) {
        vector<int> res;
        inorder(root, res);  // 子函数传入参数根节点
        
        return res[res.size()-k];
    }
};

python


class Solution:
    def kthLargest(self, root: TreeNode, k: int) -> int:
        res=[]
        def inorder(root):
            if not root:
                return 
            inorder(root.left)
            res.append(root.val)
            inorder(root.right)
        inorder(root)

        return res[len(res)-k]

在这里插入图片描述


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值