题目:给定一棵二叉搜索树,请找出其中的第K大的结点。例如下图中的二叉树,按加点数值大小顺序第三个结点的值是4。
思路:二叉搜索树的一个重要性质就是它的中序遍历是排序的,因此这道题目只需要用中序遍历算法遍历一棵二叉搜索树,就很容易找出它的第K大结点。中序遍历用递归实现比较容易,但要想清楚的是遍历到一个根结点的时候要做的是什么?中序遍历二叉树打印的时候我们在递归完左子树之后打印根结点,本题目要求的当然不是打印,如果左子结点不是要找的结点,才会访问根结点,所以访问到根结点的时候要做的操作是将k减去1,因为左子结点已经证实不是要找的结点了,排除左子结点。这个过程可以看成目标移位的过程,每移过一个结点,K减1,直到K等于1时,当前结点就是要求的结点。
BinaryTreeNode* KthNode(BinaryTreeNode* pRoot, unsigned int k)
{
if(pRoot == NULL || k == 0)
return NULL;
return KthNodeCore(pRoot, k);
}
BinaryTreeNode* KthNodeCore(BinaryTreeNode* pRoot, unsigned int& k)
{
BinaryTreeNode* target = NULL;
if(pRoot->m_pLeft != NULL)
target = KthNodeCore(pRoot->m_pLeft, k);
if(target == NULL)
{
if(k == 1)
target = pRoot;
k--;
}
if(target == NULL && pRoot->m_pRight != NULL)
target = KthNodeCore(pRoot->m_pRight, k);
return target;
}