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

最新推荐文章于 2024-10-18 10:39:20 发布

anieoo

最新推荐文章于 2024-10-18 10:39:20 发布

阅读量114

点赞数

分类专栏： Leetcode # 剑指offer 文章标签： leetcode

本文链接：https://blog.csdn.net/qq_42174306/article/details/125819703

版权

Leetcode 同时被 2 个专栏收录

400 篇文章 1 订阅

订阅专栏

剑指offer

54 篇文章 0 订阅

订阅专栏

原题链接：剑指 Offer 54. 二叉搜索树的第k大节点

solution:

二叉搜索树的中序遍历是一个升序数组，再返回第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:
    vector<int> inorder;
    int kthLargest(TreeNode* root, int k) {
        dfs(root);
        return inorder[inorder.size() - k];
    }

    void dfs(TreeNode *root) {
        if(root == NULL) return;
        dfs(root->left);
        inorder.push_back(root->val);
        dfs(root->right);
    }
};

反向中序遍历,在计算遍历结点的个数,减少消耗内存

/**
 * 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 = 0,cnt = 0;
    int kthLargest(TreeNode* root, int k) {
        dfs(root, k);
        return ans;
    }

    void dfs(TreeNode *root, int k) {
        if(root == NULL) return;

        dfs(root->right, k);
        if(++cnt == k) {
            ans = root->val;
            return;
        }
        dfs(root->left, k);
    }
};