题目
给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
限制
1 ≤ k ≤ 二叉搜索树元素个数
思路
根据题意我们需要先遍历右子树再遍历根节点再是左子树,遍历根节点时k - 1,k剪到0时本节点就是我们要找的。其中有一个剪枝操作
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int res, k;
int kthLargest(TreeNode* root, int _k) {
k = _k;
dfs(root);
return res;
}
void dfs(TreeNode* root) {
if (!root) return ;
dfs(root -> right);
k -- ;
if (k == 0) res = root -> val;
if (k > 0) dfs(root -> left);
}
};