剑指offer--二叉搜索树的第K大节点

题目描述

给定一棵二叉搜索树，请找出其中的第k大的结点。例如， （5，3，7，2，4，6，8）    中，按结点数值大小顺序第三大结点的值为4。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {}
};
*/
//思路：二叉搜索树按照中序遍历的顺序打印出来正好就是排序好的顺序。
//     所以，按照中序遍历顺序找到第k个结点就是结果。
class Solution 
{
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
         if(pRoot == NULL || k == 0) 
        return NULL; 
           
        return KthNodeCore(pRoot, k); 
    }
    TreeNode* KthNodeCore(TreeNode* pRoot, int& k) 
    { 
        TreeNode* target = NULL;  
        if(pRoot->left != NULL) 
            target = KthNodeCore(pRoot->left, k);  
        if(target == NULL) 
        { 
            if(k == 1) 
                target = pRoot;
            k--; 
        } 
        if(target == NULL && pRoot->right != NULL) 
            target = KthNodeCore(pRoot->right, k); 
        return target; 
    } 
};