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

题目链接: leetcode.

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

/*
执行用时：28 ms, 在所有 C++ 提交中击败了47.64%的用户
内存消耗：23.7 MB, 在所有 C++ 提交中击败了59.80%的用户
*/
class Solution {
	vector<int> ans;
public:
	void dfs(TreeNode* root)
	{
		if(root == nullptr)
			return;
		dfs(root -> left);
		ans.push_back(root -> val);
		dfs(root -> right);
	}
    int kthLargest(TreeNode* root, int k) {
    	dfs(root);
    	int N = ans.size();
        if(k <= 0 || k > N)
            return -1;
    	return ans[N - k];
    }
};

记住了二叉搜索树中序遍历有序，却没记住：
二叉搜索树的 中序遍历倒序 为 递减序列!!!

/*
执行用时：12 ms, 在所有 C++ 提交中击败了99.12%的用户
内存消耗：23.5 MB, 在所有 C++ 提交中击败了77.27%的用户
*/
class Solution {
	int ans;
public:
	void dfs(TreeNode* root, int &k)
	{
		if(root == nullptr || k == 0)
		{
			return;
		}
		dfs(root -> right, k);
        k--;
		if(k == 0)
		{
			ans = root -> val;
			return;
		}	
		dfs(root -> left, k);
	}
    int kthLargest(TreeNode* root, int k) {
    	dfs(root, k);
    	return ans;
    }
};

为什么遍历完右子树只需要k--，右子树有几个节点不是并不确定吗？
答：先对右树遍历，当开始判断 k-- 的时候，说明已经到达右树最底层最右边那个叶子节点，是以该节点为起点，在递归回退的时候开始k--。

所以dfs里的k取的是引用，不然形参减完回去对实参完全没用惹。。