二叉树的下一个结点

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

输入只有树中某个节点，自然地，二叉树的关系也会被传入。

感想：主要是要理清楚规律。我的方法是先找到两个特殊点，一个是根节点，一个是中序最后一个点。先把这两种的情况给返回。

其余的符合以下特点：

（1）一个节点有右子树：

          返回右子树的最左点。

（2）一个节点无右子树：

           a.为其父节点的左孩子：返回父节点

           b.为其父节点的右孩子：返回父节点的父节点。

//二叉树的下一个结点
#include <iostream>

using namespace std;

struct TreeLinkNode {
	int val;
	struct TreeLinkNode *left;
	struct TreeLinkNode *right;
	struct TreeLinkNode *next;
	TreeLinkNode(int x) :val(x), left(NULL), right(NULL), next(NULL) {

	}
};

TreeLinkNode* GetNext(TreeLinkNode* pNode)
{
	if(pNode==NULL)
		return NULL;
	TreeLinkNode* pRoot=pNode;

	while(pRoot->next!=NULL)
		pRoot=pRoot->next;

	TreeLinkNode* pEnd=pRoot;
	while(pEnd->right!=NULL)
		pEnd=pEnd->right;

	if(pNode==pEnd)
		return NULL;

	if(pNode->right!=NULL)
	{
		TreeLinkNode* pR=pNode->right;
		while(pR->left!=NULL)
		{
			pR=pR->left;
		}
		return pR;
	}else
	{
		if(pNode==pNode->next->left)
		    return pNode->next;
		else
			return pNode->next->next;
	}


}

void main()
{
	TreeLinkNode no1(8),no2(6),no3(10),no4(5),no5(7),no6(9),no7(11)/*,no8(8)*/;
	no1.left=&no2;
	no1.right=&no3;
	no2.next=&no1;
	no3.next=&no1;
	no2.left=&no4;
	no2.right=&no5;
	no4.next=&no2;
	no5.next=&no2;
	no3.left=&no6;
	no3.right=&no7;
	no6.next=&no3;
	no7.next=&no3;
	//no4.left=&no8;
	//no8.next=&no4;
	
	TreeLinkNode* no=GetNext(&no5);
	cout<<no->val;
	system("pause");

}