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

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

题目

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

思路

 按照二叉搜索树的中序遍历倒序的第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 res;
    int kthLargest(TreeNode* root, int k) {
        dfs(root,k);
        return res;
    }

    void dfs(TreeNode *root, int &k)
    {

        // 中序遍历 倒序：右节点->根结点->左节点
        if(!root) return ;

        dfs(root->right,k);
        k--;
        if(!k) res = root->val;  // 当k等于0   说明找到目标节点  无需继续遍历
        dfs(root->left,k);// 左
    }

};