二叉搜索树的第K大节点

题目描述

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

分析

对二叉搜索树进行逆向中序遍历（先右再左），则遍历序列是降序排序的，因此中序遍历一颗二叉搜索树，可以很容易的得到它的第k大的节点。使用一个计数器变量，每遍历一个节点，计数器加1，当计数器的值等于k时，root节点即为所求节点。

解法一：保存遍历过的节点，计算其数量。

public class Solution {
    vector<TreeNode> vec;
     TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(k ==  ) return null;
               zhongxu(pRoot);
        if(k > vec.size()) return null;
        return vec.get(k-);
    }
     void zhongxu(TreeNode pRoot){
                if(pRoot == null) return ;
                zhongxu(pRoot.right);
                vec.push_back(pRoot);
                zhongxu(pRoot.left);
    }  
 
} 

解法二：利用递归的返回值，判断是否结束。

class Solution {
    int count = ;
public:
    TreeNode* KthNode(TreeNode* pRoot, unsigned int k)
    {
        if(pRoot){ 
                TreeNode *ret = KthNode(pRoot->right, k);   //如果是叶子节点，到这里返回。然后记数
                if(ret) return ret;
                count++;   //this ++ because root is not null
                if(count == k)
                    return pRoot;
                ret = KthNode(pRoot->left,k);  //路径是先右后左
                if(ret) return ret;
        }
        return nullptr;
    }
};