【题目】
给定一棵二叉搜索树,请找出其中第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;
}
};