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

cx_cs

于 2022-03-09 17:07:30 发布

阅读量47

点赞数

分类专栏：树文章标签：深度优先 leetcode 数据结构

本文链接：https://blog.csdn.net/cx_cs/article/details/123382176

版权

树专栏收录该内容

53 篇文章 0 订阅

订阅专栏

该博客介绍了如何使用深度优先搜索(DFS)策略找到二叉树中的第k个最大元素。通过从右到左遍历节点并递归地处理子树，可以在O(n)的时间复杂度内找到目标值。代码实现中定义了一个`Solution`类，包含一个全局变量`ans`用于存储结果，并提供了一个`dfs`辅助函数进行实际的DFS操作。

摘要由CSDN通过智能技术生成

要求：rt
思路：右根左

/**
 * 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 ans;
    void dfs(TreeNode* root,int &k){
        if(!root)return;
        if(k==0)return;
        dfs(root->right,k);
        --k;
        if(k==0)ans=root->val;
        dfs(root->left,k);
    }
    int kthLargest(TreeNode* root, int k) {
        dfs(root,k);
        return ans;
    }
};