问题描述
题目:给定一棵二叉搜索树,请找出其中第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;
}