ybt1366 二叉树输出

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;
}