二叉搜索树的第K大节点

题目描述

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

分析

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

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

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

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

class Solution {
    int count = 0;
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;
    }
};