二叉树输出(btout)

二叉树输出(btout)

链接：http://ybt.ssoier.cn:8088/problem_show.php?pid=1366
时间限制: 1000 ms         内存限制: 65536 KB
 

【题目描述】

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

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

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

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

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

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

 

【输入】

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

【输出】

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

【输入样例】

ABCDEFG
CBDAFEG

【输出样例】

AAAA
BB
C
D
EE
F
G

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<map>
#include <algorithm>
using namespace std;
vector <char> pre,in,post;
map <int,char> M;
int c[105];
int i,up;
void rec(int l,int r,int k)
{
    if(l>=r)
    {
        c[k/2]=1;return;
    }
    int m=distance(in.begin(),find(in.begin(), in.end(), pre[i++]));    
    M[k]=in[m];
    rec(l,m,k*2);
    rec(m+1,r,k*2+1);
    up=max(up,k);
}
int js(int k)
{
    if(k>up)return 0;
    if(c[k])return c[k];
    return c[k]=js(2*k)+js(2*k+1);
}
void print(int k)
{
    if(!c[k])return ;
    for(int j=1;j<=c[k];j++)
        printf("%c",M[k]);
    cout<<endl;
    print(2*k);
    print(2*k+1);
}
int main()
{
    string s1,s2;
    cin>>s1>>s2;
    for(int i=0;i<s1.size();i++)
        pre.push_back(s1[i]);
    for(int i=0;i<s2.size();i++)
        in.push_back(s2[i]);
    rec(0,s1.size(),1);
    for(int i=0;i<s1.size();i++)
    js(1);
    print(1);
    
    cout<<endl;
}