每日一练(26):二叉搜索树的第k大节点

简介: 给定一棵二叉搜索树,请找出其中第 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


限制:


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


来源:力扣(LeetCode)


链接:https://leetcode-cn.com/probl...


方法一:vector递归


class Solution {
public:
    vector<int> ans;
    void dfs(TreeNode* root) {
        if (!root) {
            return;
        }
        dfs(root->left);
        ans.push_back(root->val);
        dfs(root->right);
    }
    int kthLargest(TreeNode* root, int k) {
        dfs(root);
        return ans[ans.size() - k];
    }
};


方法二:变量递归


把原本是 “左中右” 的中序遍历改成 “右中左” 的反向中序遍历

维护两个变量 count 和 res 即可。count 用来计数我们在降序的数字序列中走了几位,当走了 k 位时,就让 res 等于当前的 root -> val,然后退出 inorder 函数即可


class Solution {
public:
    // 二叉搜索树的中序遍历是递增序列 
    int  count, res;
    void dfs(TreeNode* root, int k){
        if (root == nullptr) {
            return;
        }
        dfs(root->right, k);
        count++;
        if (count == k) {
            res = root->val;
            return;
        }
        dfs(root->left, k);
    }
    int kthLargest(TreeNode* root, int k) {
        dfs(root, k);
        return res;
    }
};
目录
相关文章
|
7月前
|
存储
二叉树进阶之二叉搜索树
二叉树进阶之二叉搜索树
51 0
|
C++
二叉树进阶 - (C++二叉搜索树的实现)
二叉树进阶 - (C++二叉搜索树的实现)
59 1
|
5月前
|
存储 C++
【C++】二叉树进阶之二叉搜索树(下)
【C++】二叉树进阶之二叉搜索树(下)
35 4
|
5月前
|
Java 编译器 C++
【C++】二叉树进阶之二叉搜索树(上)
【C++】二叉树进阶之二叉搜索树(上)
39 3
|
7月前
|
算法
二叉树刷题记(六-二叉搜索树的第k大节点)
二叉树刷题记(六-二叉搜索树的第k大节点)
|
7月前
|
算法 DataX
二叉树(中)+Leetcode每日一题——“数据结构与算法”“剑指Offer55-I. 二叉树的深度”“100.相同的树”“965.单值二叉树”
二叉树(中)+Leetcode每日一题——“数据结构与算法”“剑指Offer55-I. 二叉树的深度”“100.相同的树”“965.单值二叉树”
Leetcode1038. 从二叉搜索树到更大和树(每日一题)
Leetcode1038. 从二叉搜索树到更大和树(每日一题)
|
算法
代码随想录算法训练营第二十一天 | LeetCode 235. 二叉搜索树的最近公共祖先、701. 二叉搜索树中的插入操作、450. 删除二叉搜索树中的节点
代码随想录算法训练营第二十一天 | LeetCode 235. 二叉搜索树的最近公共祖先、701. 二叉搜索树中的插入操作、450. 删除二叉搜索树中的节点
66 0
剑指offer-7.二叉树的下一个节点
剑指offer-7.二叉树的下一个节点
40 1
|
C++
剑指Offer - 面试题8:二叉树的下一个节点
剑指Offer - 面试题8:二叉树的下一个节点
64 0