1366：二叉树输出(btout)

上一次的代码有些小错误，所以呢我重写了一遍

【题目描述】

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

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

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

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

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

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

【输入】

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

【输出】

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

【输入样例】

ABCDEFG
CBDAFEG

【输出样例】

AAAA
BB
C
D
EE
F
G

代码：

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <map>
#include <set>
#include <queue>
#include <stack>
#define sf(a) scanf("%d\n",&a)
#define pf(a) printf("%.2lf\n",a)
#define pfc(a) printf("%c",a)
#define pi acos(-1.0)
#define E 1e-8
#define ms(a) memset(a,0,sizeof a)
#define min_ms(a) memset(a,-1,sizeof a)
#define max_ms(a) memset(a,0x3f,sizeof a)
#define rep(a,b,c) for(int a=b;a<=c;a++)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef struct node;
typedef node *tree;
const int inf=0x3f3f3f3f;
const int idata=2e6+5;
const int mod=1e9+7;
struct node
{
    char data;
    node *lchild;
    node *rchild;
};
string s1,s2;
int cnt[idata];
 
int calc(int l1,int r1,int l2,int r2)
{
    if(l1==r1) return cnt[l1]=1;
 
    int i;
    for(i=l2;i<=r2;i++)
    {
        if(s1[l1]==s2[i]) break;
    }
 
    int p=i-l2;
    if(i>l2) cnt[l1]+=calc(l1+1,p+l1,l2,i);
    if(i<r2) cnt[l1]+=calc(l1+p+1,r1,i+1,r2);
 
    return cnt[l1];
}
 
signed main()
{
    while(cin>>s1>>s2)
    {
        calc(0,s1.length()-1,0,s2.length()-1);
        for(int i=0;i<s1.length();i++)
        {
            for(int j=1;j<=cnt[i];j++)
            {
                cout<<s1[i];
            }
            cout<<endl;
        }
        exit(0);
    }
    return 0;
}