ybt1366 二叉树输出
时空限制 1000ms/128MB
【题目描述】
树的凹入表示法主要用于树的屏幕或打印输出,其表示的基本思想是兄弟间等长,一个结点的长度要不小于其子结点的长度。二叉树也可以这样表示,假设叶结点的长度为1,一个非叶结点的长度等于它的左右子树的长度之和。
一棵二叉树的一个结点用一个字母表示(无重复),输出时从根结点开始:
每行输出若干个结点字符(相同字符的个数等于该结点长度),
如果该结点有左子树就递归输出左子树;
如果该结点有右子树就递归输出右子树。
假定一棵二叉树一个结点用一个字符描述,现在给出先序和中序遍历的字符串,用树的凹入表示法输出该二叉树。
【输入】
两行,每行是由字母组成的字符串(一行的每个字符都是唯一的),分别表示二叉树的先序遍历和中序遍历的序列。
【输出】
行数等于该树的结点数,每行的字母相同。
【输入样例】
ABCDEFG CBDAFEG
【输出样例】
AAAA BB C D EE F G
代码
#include<iostream>
#include<string>
using namespace std;
struct node{
char data;
int num;
node *lch,*rch;
};
void create(node *&root,string s1,string s2){
if (s1=="") { root=NULL; return; }
root = new node;
root->data = s1[0];
int len=s1.size(),pos=s2.find(s1[0]);
create(root->lch,s1.substr(1,pos),s2.substr(0,pos));
create(root->rch,s1.substr(pos+1,len-pos-1),s2.substr(pos+1,len-pos-1));
}
void find(node *root){
if (root->lch==NULL && root->rch==NULL) { root->num=1; return; }
if (root->lch!=NULL && root->rch==NULL){
find(root->lch);
root->num = root->lch->num;
return;
}
if (root->lch==NULL && root->rch!=NULL){
find(root->rch);
root->num = root->rch->num;
return;
}
find(root->lch);
find(root->rch);
root->num = root->lch->num+root->rch->num;
}
void print(node *root){
if (root){
for (int i=1; i<=root->num; i++) cout<<root->data;
cout<<endl;
print(root->lch);
print(root->rch);
}
}
int main(){
string s1,s2;
cin>>s1>>s2;
node *root(NULL);
create(root,s1,s2);
find(root);
print(root);
return 0;
}