『剑指Offer』54. 二叉搜索树的第k大节点

一枚方糖

于 2021-11-15 19:59:10 发布

阅读量1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：剑指Offer 文章标签：深度优先算法 leetcode

本文链接：https://blog.csdn.net/xg987599519/article/details/121342022

剑指Offer 专栏收录该内容

30 篇文章

订阅专栏

文章目录

题目
思路
代码

题目

在这里插入图片描述

思路

遍历树 val 放入 vector ，然后排序输出第 K 大。

代码

/**
 * 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 kthLargest(TreeNode* root, int k) {
        vector<int> res;
        if (root == NULL) return 0;
        dfs(root,res);
        sort(begin(res), end(res));
        return res[size(res) - k];
    }
    void dfs(TreeNode* root, vector<int> &res)
    {   
        if (root == NULL) return ;
        res.push_back(root->val);
        dfs(root->left,res);
        dfs(root->right,res);
    }
};