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

分析：这道题比较简单，二叉搜索树的中序遍历序列是递增序列。我们对中序遍历做下修改，改成先遍历右子树，然后访问根节点，最后访问左子树，这样得到的中序遍历序列就是递减序列，我们在完成第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:
    int kthLargest(TreeNode* root, int k) {
        int res=middlebianli(root,k);
        return res;
    }
    int middlebianli(TreeNode *root,int &k)
    {

        if(root==NULL) return 0;
        // 先访问右子树
        int right;
        if(root->right!=NULL) right=middlebianli(root->right,k);
        // 访问完右子树之后，如果k为0，说明第k个节点出现在右子树中，返回右子树的遍历结果即可
        if(k==0) return right;
        // 访问根节点，并使k--
        k--;
        // 访问完根节点之后，如果k为0，说明根节点就是第k个节点，直接返回根节点即可
        if(k==0) return root->val;
        // 最后访问左子树
        int left;
        if(root->left!=NULL) left=middlebianli(root->left,k);
        // 访问完左子树之后，如果k为0，说明第k个节点出现在左子树中，返回左子树的遍历结果即可
        if(k==0) return left;
        // 没有结果就返回0
        return 0;
    }
};