剑指 Offer 54. 二叉搜索树的第k大节点
给定一棵二叉搜索树,请找出其中第 k 大的节点的值。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4
示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4
使用先序遍历,用一个容器存放个节点数值,然后对容器使用lambda 和 sort 进行从大到小排序。
class Solution {
public:
void order(TreeNode* root, vector<int>& res)
{
if(root == nullptr)
return ;
res.push_back(root->val);
order(root->left, res);
order(root->right, res);
}
int kthLargest(TreeNode* root, int k) {
vector<int> vec;
order(root, vec);
sort(vec.begin(),vec.end(),[](int x, int y){return x>y;});
return vec[k-1];
}
};