剑指 Offer 54. 二叉搜索树的第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:
    vector<int> res;

    int kthLargest(TreeNode* root, int k) {

        // 二叉搜索树，中序遍历就是递增序列
        dfs(root);   
        reverse(res.begin(), res.end());
        return res[k-1];

    }

    void dfs(TreeNode* root){
        // 中序
        if(root==NULL) return;

        dfs(root->left);
        res.push_back(root->val); // visit
        dfs(root->right);
    }

};

上面不是最优解。因为二叉搜索树的中序遍历（左-根-右）是一个递增的过程了，那么（右-根-左）就是一个递减的过程。所以要找到第k大的结点，可以按照（右-根-左）进行遍历，倒计时数数，k--，k减为0了就是第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){   // k加上引用，每次访问共享同一个k的值
        // 中序
        if(root==NULL || k<=0) return;

        dfs(root->right, k);
        // visit
        k--;
        if(k==0){
            res = root->val;
            return;
        }

        dfs(root->left, k);
    }

};