题目:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
分析:这个题目需要我们考虑多种情况,1.考虑有没有右孩子;2.有右孩子,考虑右孩子有没有左孩子;3.没有右孩子,考虑是否存在一个祖父节点,且该节点是其父节点的左孩子;以上情况都考虑清楚了,题目自然就解决了。
程序:
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if (pNode->right&&!pNode->right->left) return pNode->right; //有右孩子但没右孩子的左孩子;
else if (pNode->right&&pNode->right->left) return pNode->right->left; //有右孩子且有右孩子的左孩子;
else { //如果没有右孩子,则判断是否存在最近的一个祖父,且是其父节点的左孩子;
TreeLinkNode *p = pNode;//指针初始化,只有3种情况,1.赋为NULL;2.赋一个同类型的变量地址;2.开辟一段新的内存,其指向新内存的首地址;
while (p->next&&p->next->right == p)
p = p->next;
if (p->next&&p->next->left == p) return p->next;
else return NULL;
}
}