二叉树的下一个结点

题目描述

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

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

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

其余的符合以下特点：

（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");

}