NYOJ-756重建二叉树

点击打开题目链接

1.后序序列的最后一个元素是二叉树的根节点；

2.在中序序列中找到根节点的位置，则中序序列中根节点左边的元素是该根节点的左子树，右边的元素是根节点的右子树；

3.由左子树的节点  在后序序列中找到对应的序列，则所得序列的最后一个元素就是左子树的根节点（右子树构建的方法与此相同）。

所以有递归的方法很方便的就能重建二叉树并得到先序序列。

#include <iostream>
#include <cstdio>
#include <cstring>
//#define LOCAL
using namespace std;
typedef char Elemtype;
//----------------------------二叉树结点定义
class BTNode
{
public:
BTNode(Elemtype _data='#',BTNode *_lchild=NULL,BTNode *_rchild=NULL):data(_data),lchild(_lchild),rchild(_rchild){}

void Setdata(Elemtype _data) {data=_data;}

void Setlchild(BTNode *_lchild){lchild=_lchild;}

void Setrchild(BTNode *_rchild){rchild=_rchild;}

Elemtype Getdata() {return data;}

BTNode *Getlchild() {return lchild;}

BTNode *Getrchild() {return rchild;}

private:
Elemtype data;
BTNode *lchild,*rchild;
};

//二叉树功能定义
class BTFun
{
public:
BTFun(BTNode *_root=NULL):root(_root){}
~BTFun() {delete root;}

void Setroot(BTNode *_root) {root=_root;}
BTNode *Getroot(){return root;}

void Preorder(BTNode *t); //先序遍历

int Find(char ch,char in[],int in_s,int in_e); //寻找位置
BTNode *Rebuild(char pre[],char in[],int pre_s,int pre_e,int in_s,int in_e); //重建二叉树

private:
BTNode *root;
};

void BTFun::Preorder(BTNode *t) //得到先序序列并输出
{
if(t!=NULL)
{
cout<<t->Getdata();
Preorder(t->Getlchild());
Preorder(t->Getrchild());
}
}

int BTFun::Find(char ch,char in[],int in_s,int in_e) //返回字符ch在数组in中的位置
{
int i;
for(i=in_s;i<=in_e;i++)
if(in[i] == ch) break;
return i;
}

BTNode* BTFun::Rebuild(char post[],char in[],int post_s,int post_e,int in_s,int in_e)
{
if(in_s>in_e) return NULL;
BTNode *temp=new BTNode(post[post_e]);
int mid=Find(post[post_e],in,in_s,in_e);
temp->Setlchild(Rebuild(post,in,post_s,post_s+mid-in_s-1,in_s,mid-1)); //构建左子树
temp->Setrchild(Rebuild(post,in,post_s+mid-in_s,post_e-1,mid+1,in_e)); //构建右子树
return temp; //返回根节点
}

int main()
{
#ifdef LOCAL
freopen("Input2.txt","r",stdin);
freopen("Output2.txt","w",stdout);
#endif
char post[30],in[30];
while(~scanf("%s%s",post,in))
{
BTFun T;
int len=strlen(post);
BTNode *t=T.Rebuild(post,in,0,len-1,0,len-1);
T.Preorder(t);
cout<<endl;
}
return 0;
}