剑指Offer54-二叉搜索树的第k大节点-easy

试题链接

题目描述：

给定一棵二叉搜索树，请找出其中第k大的节点。

例如输入：

输入: root = [3,1,4,null,2], k = 1
   3
  / \
 1   4
  \
   2
输出: 4

输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 4

数据范围：

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

解题思路：

二叉搜索树，左子树所有结点比根结点小，右子树所有结点比根结点大，左右子树所有结点都满足此规则。
二叉搜素树的中序遍历序列是一个递增序列。但是此题要的是第k大的结点。所以要求中序遍历的倒序。
递归 遍历顺序 ： 1 .右子树 2. 根结点 3.左子树
用一个num记录当前是第几个值，当num == k时可以提前结束递归。

AC代码(c++)

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