【题解-信息学奥赛一本通】1366：二叉树输出(btout)

题目：1366：二叉树输出(btout)

题目描述

树的凹入表示法主要用于树的屏幕或打印输出，其表示的基本思想是兄弟间等长，一个结点的长度要不小于其子结点的长度。二叉树也可以这样表示，假设叶结点的长度为1，一个非叶结点的长度等于它的左右子树的长度之和。

一棵二叉树的一个结点用一个字母表示（无重复），输出时从根结点开始：

每行输出若干个结点字符（相同字符的个数等于该结点长度），

如果该结点有左子树就递归输出左子树；

如果该结点有右子树就递归输出右子树。

假定一棵二叉树一个结点用一个字符描述，现在给出先序和中序遍历的字符串，用树的凹入表示法输出该二叉树。

输入

两行，每行是由字母组成的字符串（一行的每个字符都是唯一的），分别表示二叉树的先序遍历和中序遍历的序列。

输出

行数等于该树的结点数，每行的字母相同。

时空限制

1s / 64MB

样例输入

ABCDEFG
CBDAFEG

样例输出

AAAA
BB
C
D
EE
F
G

代码

#include<bits/stdc++.h>
using namespace std;
const int N=26+10;
string preorder,inorder;
unordered_map<char,int> pos,path;
vector<char> q;
int dfs(int pl,int pr,int il,int ir){
char root=preorder[pl];
int m=pos[root];
int l=0,r=0;
q.push_back(root);
if(m>il){
path[root]+=dfs(pl+1,pl+m–il,il,m–1);
l=1;
}
if(m<ir){
path[root]+=dfs(pl+m–il+1,pr,m+1,ir);
r=1;
}
if(l+r==0){
path[root]=1;
return 1;
}
return path[root];
}
int main(){
cin>>preorder>>inorder;
for(int i=0;i<inorder.size();i++) pos[inorder[i]]=i;
dfs(0,preorder.size()–1,0,inorder.size()–1);
for(int i=0;i<preorder.size();i++){
while(path[preorder[i]]—) cout<<preorder[i];
puts("");
}
return 0;
}

结果