Fourth Exercise：二叉树的下一个结点

题目要求：给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。【来源于剑指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，函数只有一个参数，无法实现在一个二叉树里面查找某个节点的功能；同时，也不知道如何将要查找的节点表示出来，左右子树的指针分别指向哪里呢？？目前，没有想出相应的解决方法~~