[题解]剑指 Offer 54. 二叉搜索树的第k大节点(C++)

题目

给定一棵二叉搜索树，请找出其中第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/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

二叉搜索树中序遍历是单调递增，变一下中序遍历（先右子树后左子树）就是递减数组，遍历的时候带上k，每次过一个节点就--k，当k为0时就得到了需要的值。
时间复杂度O(n)，空间复杂度O(n)。

代码

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

private:
    void reinorder(TreeNode *node, int &k)
    {
        if(!node) return;
        reinorder(node->right, k);
        if(k == 0) return;
        if(--k == 0)
        {
            ans = node->val;
        }
        reinorder(node->left, k);
    }
};