【简单】面试题54. 二叉搜索树的第k大节点

643 篇文章 5 订阅

【题目】
给定一棵二叉搜索树,请找出其中第k大的节点。
来源:leetcode
链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/
【示例】
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
【思路】
1、 二叉树的中序遍历的拟状态,正常二叉树的中序遍历是LNR,而其逆状态是RNL
2、根据题目要求可知,需要找到第k大的数字并返回,由于BST二叉搜索树的中序遍历是从小到大的在没有重复元素的时候可以看作一个递增数列,其逆中序遍历是从大到小的递减数列。所以此题找第k大的数字比较适合用逆中序遍历。
【代码】

class Solution {
public:
    int cnt=0;
    int kthLargest(TreeNode* root, int k) {
        //RNL 右根左 中序遍历
        stack<TreeNode*> s;
        TreeNode* treeNode = root;
        while(treeNode!=NULL || !s.empty()){
            while(treeNode != NULL){
                s.push(treeNode);
                treeNode = treeNode->right;
            }
            if(!s.empty()){
                treeNode = s.top();
                s.pop();
                cnt++;
                if(cnt==k)
                    return treeNode->val;
                treeNode = treeNode->left;
            }
        }
        return 0;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值