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

方法一：

由于是有序的平衡二叉树，因此是中序。根据dfs中序迭代求得递增排序后，再反转就可求得结果。

class Solution {
public:
    vector<int> tmp;
    int kthLargest(TreeNode* root, int k) {
        // if(root != NULL) return 
        dfs(root);
        reverse(tmp.begin(), tmp.end());
        return tmp[k -1];
    }
    void dfs(TreeNode* root) {
        if(root == NULL) return;
        dfs(root->left);
        tmp.emplace_back(root->val);
        dfs(root->right);
    }
};

时间28 ms

内存23.8 MB

方法二：

改进：直接将中序遍历改成右->根->左，这样既省去了开辟额外的数组，又省去了最后的reverse的O(n)算法复杂度。

注意：dfs迭代是，传递的k是引用！！

class Solution {
public:
    int res;
    int kthLargest(TreeNode* root, int k) {
        dfs(root, k);
        return res;
    }
    void dfs(TreeNode *root, int &k) {
        if(root == NULL || k <= 0) return;
        dfs(root->right, k);
        k --;
        if(k == 0) {
            res = root->val;
            return;
        }
        dfs(root->left, k);
    }
};

时间20 ms

内存23.4 MB