【二叉樹】二叉搜索樹的第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;
}