题目要求:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。【来源于剑指offer】
思路整理:分析二叉树的中序遍历顺序,可以分为以下情况:
- 该节点为父亲的左子树:
- 该节点有右子树:下一节点为右子树的最左边
- 该节点无右子树:下一节点为父节点
- 该节点为父亲的右子树:
- 该节点有右子树:下一节点为右子树的最左边
- 该节点无右子树:下一节点为父节点的父节点,直到父节点处于左子树位置
将上述分析情况进行整合:
- 该节点有右子树:下一节点为右子树的最左边
- 该节点无右子树:
- 该节点为父亲的左子树:下一节点为父节点
- 该节点为父亲的右子树:下一节点为父节点的父节点,直到父节点处于左子树位置
代码实现: 依照上述思路实现,得到了如下错误代码(找父节点为左子树、右节点的最左节点的循环有问题)
treelinknode* getnext(treelinknode* pnode)
{
treelinknode* parent = pnode->next;//未考虑根节点、空节点的情况
treelinknode* nextnode;
if (pnode->right == null)
{
if(pnode == parent->left)
nextnode = parent;
else
{
do
{
nextnode = parent;
parent = parent->next;
} while (nextnode != parent->left);
}
}
else do
{
nextnode = pnode->left;//右子树的最左子树,右子树不应该在循环体内
pnode = pnode->left;
} while (pnode->left != null);
return nextnode;
}
思路误区:查看参考程序,对比代码,分析思考:我采取的方法是,设置了一个变量,各种情况下分别赋值,最后一次性返回该变量值。
此种方法容易出现问题:
- 需要保证只进入一种情况,以防止执行完第一种情况后又继续执行第二种情况;
- 未考虑到该节点为空节点、根节点、中序遍历的最后一个节点等各种情况,应将这些情况都要考虑进去,写入代码,不能仅用if else分为两种情况。
大佬们采用的排除法完美地解决了以上问题:有返回值的各个情况罗列出来,不符合要求的全部返回为空。
代码修正
TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
if (pNode == NULL)return NULL;
if (pNode->right != NULL)
{
pNode = pNode->right;
while (pNode->left != NULL)
pNode = pNode->left;
return pNode;//每种情况返回一次,不用设定一个变量,最后一次性返回全部
}
//遍历的最后一个元素的情况较复杂,无法判断,所以最后返回空比较好,有节点返回的情况下直接返回节点
while (pNode->next != NULL)
{
TreeLinkNode* parent = pNode->next;
if (pNode == parent->left)
return parent;
pNode = pNode->next;
}
return NULL;
}
拓展思考
- 本来想在VS里写出完整工程,调试实现,But,函数只有一个参数,无法实现在一个二叉树里面查找某个节点的功能;同时,也不知道如何将要查找的节点表示出来,左右子树的指针分别指向哪里呢??目前,没有想出相应的解决方法~~