其实整个算法用到的是中序遍历的思想,只不过是先遍历右子树,然后遍历根节点,再遍历右子树,每次遍历一个节点的时候thK减1,当thK为0时,就是需要找的节点
Node* findKthNodeImp(Node *pNode, int& thK)
{
if ((pNode == NULL)|| (thK <= 0))
{
return NULL;
}
Node *pTmp = NULL;
pTmp = findKthNodeImp(pNode->rightChild, thK);
if (pTmp)
{
return pTmp;
}
thK--;
if (thK == 0)
{
return pNode;
}
else
{
pTmp = findKthNodeImp(pNode->leftChild, thK);
if (pTmp)
{
return pTmp;
}
}
return NULL;
}