【二叉树】二叉搜索树的第k个结点

问题描述

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

分析问题

由于是二叉搜索树，所以中序遍历该树，得到一个递增的序列。
求第k大的结点，根据中序遍历，遍历k个数。

Node* KthNode(Node* pRoot,int k)
{
    if (pRoot == NULL || k == 0)
        return NULL;
    return KthNodeCore(pRoot, k);
}

Node *KthNodeCore(Node *pRoot, int& k)
{
    //返回的第k个节点
    Node* KNode = NULL;
    //先找到最左边的节点，
    if (pRoot->left != NULL)
        KNode = KthNodeCore(pRoot->left, k);

    //遍历根节点，--k，直到k==1，返回当前遍历结点
    if (KNode == NULL)
    {
        if (k == 1)
            KNode = pRoot;
        --k;
    }
    //如果根节点不是k结点且右子树存在，遍历右子树。
    if (KNode == NULL && pRoot->right != NULL)
        KNode = KthNodeCore(pRoot->right, k);
    return KNode;
}