leetcode：剑指 Offer 54. 二叉搜索树的第 k 大节点（C++）

最新推荐文章于 2022-06-19 19:26:02 发布

wangqingchuan92

最新推荐文章于 2022-06-19 19:26:02 发布

阅读量357

点赞数

分类专栏： LeetCode刷题记录文章标签： leetcode c++ 深度优先

本文链接：https://blog.csdn.net/wangqingchuan92/article/details/122664068

版权

LeetCode刷题记录专栏收录该内容

274 篇文章 27 订阅

订阅专栏

题目描述：
给定一棵二叉搜索树，请找出其中第 k 大的节点的值。

示例 1:

输入: root = [3,1,4,null,2], k = 1

输出: 4

示例 2:

输入: root = [5,3,6,2,4,null,null,1], k = 3

输出: 4

限制：

1 ≤ k ≤ 二叉搜索树元素个数

作者：Krahets
链接：https://leetcode-cn.com/leetbook/read/illustration-of-algorithm/58df23/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
解答：

/**
 * 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) {
        this->k = k;
        dfs(root);
        return res;
    }
private:
    int res, k;
    void dfs(TreeNode* root) {
        if(root == nullptr) return;
        dfs(root->right);
        if(k == 0) return;
        if(--k == 0) res = root->val;
        dfs(root->left);
    }
};