第十七天剑指 Offer 54. 二叉搜索树的第k大节点

剑指 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];

    }
};
上一篇:C++ vector类的用法整理


下一篇:计算几何入门