基于中序遍历找到一个结点的后继结点

时间:2022-10-02 11:27:33
题目: 基于中序遍历找到一个结点的后继结点。
分析: 首先明确中序遍历,顺序为:左--->根----->右 假设当前结点为p。 有两种情况: 1.当p有右子树时,那么其右子树的最左结点即为所求:

基于中序遍历找到一个结点的后继结点

2.当p没有右子树时,有下面两种情况:

基于中序遍历找到一个结点的后继结点基于中序遍历找到一个结点的后继结点

沿着p向上找,如果p的父结点的左孩子是p,那么该父结点即为所求,否则继续向上找。
代码:
/*
找到中序遍历下一个结点的后继结点
by Rowandjj
2014/8/19
*/
#include<iostream>
using namespace std;
typedef struct _NODE_
{
int data;
struct _NODE_ *left;
struct _NODE_ *right;
struct _NODE_ *parent;
}Node,*pNode,*pTree;
//找到中序遍历时,p的后继结点
pNode after(pNode p)
{
if(p == NULL)
{
return NULL;
}
if(p->right != NULL)//存在右子树
{//找到右子树中的最左结点
pNode pTemp = p->right;
while(pTemp->left)
{
pTemp = pTemp->left;
}
return pTemp;
}else//不存在右子树
{//那么找到该结点的父结点,若当前结点是父结点的左孩子那么父结点即为所求
//否则继续向上寻找
pNode pParent = p->parent;
while(pParent && pParent->right == p)
{
p = pParent;
pParent = p->parent;
}
return pParent;
}
}
//构建
void create(pTree *pRoot,pNode pParent)
{
int data;
cin>>data;
if(data == -1)
{
return;
}
*pRoot = (pNode)malloc(sizeof(Node));
if(*pRoot == NULL)
{
exit(-1);
}
(*pRoot)->data = data;
(*pRoot)->left = NULL;
(*pRoot)->right = NULL;
(*pRoot)->parent = pParent;
create(&(*pRoot)->left,*pRoot);
create(&(*pRoot)->right,*pRoot);
}
//中序遍历
void display(pTree pRoot)
{
if(pRoot == NULL)
{
return;
}
if(pRoot->left != NULL)
{
display(pRoot->left);
}
cout<<pRoot->data<<" ";
if(pRoot->right != NULL)
{
display(pRoot->right);
}
}
int main()
{
pTree pTree = NULL;
create(&pTree,NULL);
// pNode p1 = pTree->left->right;
// cout<<p1->data<<endl;
// cout<<p1->parent->data<<endl;
//cout<<after(pTree)->data;
//cout<<after(pTree->left->right)->data<<endl;
cout<<after(pTree->left)->data<<endl;
display(pTree);
return 0;
}


相关文章