二叉樹輸出(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;
}