欢迎访问~原文出处——博客园-zhouzhendong
去博客园看该题解
题目传送门 - 51Nod1832
题意概括
对于给定的一个二叉树的先序遍历和后序遍历,输出有多少种满足条件的二叉树。
两棵二叉树不同当且仅当对于某个x,x的左儿子编号不同或x的右儿子编号不同。
题解
我们发现,如果两棵二叉树先后序遍历相同,但是形态不同,只可能是某些节点,只有一个子节点,这个子节点在左边和右边都可以的情况。
那么只需要统计这样的节点个数,然后2^tot,高精度即可。
代码
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int N=10000+5;
int n,tot,a[N],b[N],pa[N],pb[N];
int d,x[1005];
void dfs(int la,int ra,int lb,int rb){
if (la==ra)
return;
la++,rb--;
int p=pb[a[la]];
if (p==rb){
tot++;
dfs(la,ra,lb,rb);
return;
}
int s=p-lb+1;
dfs(la,la+s-1,lb,lb+s-1);
dfs(la+s,ra,lb+s,rb);
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&a[i]),pa[a[i]]=i;
for (int i=1;i<=n;i++)
scanf("%d",&b[i]),pb[b[i]]=i;
tot=0;
dfs(1,n,1,n);
memset(x,0,sizeof x);
d=1,x[1]=1;
while (tot--){
int mod=1e9;
for (int i=1;i<=d;i++)
x[i]<<=1;
for (int i=1;i<=d;i++)
x[i+1]+=x[i]/mod,x[i]%=mod;
if (x[d+1])
d++;
}
printf("%d",x[d]);
for (int i=d-1;i>0;i--)
printf("%09d",x[i]);
return 0;
}