二叉树凹入法输出

题目描述

      树的凹入表示法主要用于树的屏幕或打印输出，其表示的基本思想是兄弟间等长，一个结点要不小于其子结点的长度。二叉树也可以这样表示，假设叶结点的长度为1，一个非叶结点的长并等于它的左右子树的长度之和。
      一棵二叉树的一个结点用一个字母表示（无重复），输出时从根结点开始：
      每行输出若干个结点字符（相同字符的个数等于该结点长度），
      如果该结点有左子树就递归输出左子树；
      如果该结点有右子树就递归输出右子树。
      假定一棵二叉树一个结点用一个字符描述，现在给出先序和中序遍历的字符串，用树的凹入表示法输出该二叉树。

 

输入

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

输出

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

样例输入

ABCDEFG
CBDAFEG

样例输出

AAAA
BB
C
D
EE
F
G

#include<iostream>
#include<cstring>
using namespace std;
const int maxn=100001;
char a[maxn],b[maxn],c[maxn];
int dfs(int l1,int r1,int l2,int r2)
{
    if(l1==r1)
    {
        c[l1]=1;
        return c[l1];
    }
    int i;
    for(i=l2;i<=r2;i++)//找出根在中序序列中的位置 
        if(a[l1]==b[i])
            break;
    if(l2<i)//如果有左子树，递归寻找孩子，累加，右子树同理 
        c[l1]+=dfs(l1+1,(i-1-l2)+(l1+1),l2,i-1);
    if(i<r2)
        c[l1]+=dfs(r1-(r2-(i+1)),r1,i+1,r2);
    return c[l1];//返回每个节点的孩子个数 
}
int main()
{
    cin>>a>>b;
    int lena=strlen(a);
    int lenb=strlen(b);
    dfs(0,lena-1,0,lenb-1);
    for(int i=0;i<lena;i++)
    {
        for(int j=1;j<=c[i];j++)
            cout<<a[i];
        cout<<endl;
    }
    return 0;
}