51nod 1832 先序遍历与后序遍历（DFS）

1832 先序遍历与后序遍历

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

对于给定的一个二叉树的先序遍历和后序遍历，输出有多少种满足条件的二叉树。
两棵二叉树不同当且仅当对于某个x，x的左儿子编号不同或x的右儿子编号不同。

Input

第一行一个正整数n(3<=n<=10000)，表示二叉树的节点数，节点从1到n标号。
第二行n个整数a[i](1<=a[i]<=n)，表示二叉树的先序遍历。
第三行n个整数b[i](1<=b[i]<=n)，表示二叉树的后序遍历。

Output

输出一个整数表示有多少种方案。保证至少有1种方案。

Input示例

3
1 2 3
2 3 1

Output示例

1

Scape  (题目提供者)

Visual C++的运行时限为：1000 ms ，空间限制为：131072 KB  示例及语言说明请按这里

 允许其他 AC 的用户查看此代码，分享代码才能查看别人的代码并有机会获得勋章

题解：看了一篇百度文库，觉得讲的不错，于是照着上边的思路敲了一发。觉得在纸上画画会好理解些。

百度文库链接：https://wenku.baidu.com/view/a2a45aa0284ac850ad024261.html

#include<set>  
#include<map>         
#include<stack>                
#include<queue>  
#include<vector>        
#include<string>      
#include<math.h> 
#include<time.h>  
#include<stdio.h>                
#include<iostream>                
#include<string.h>                
#include<stdlib.h>        
#include<algorithm>       
#include<functional>        
using namespace std;
typedef long long ll;
#define inf 1000000000           
#define mod 1000000000                
#define maxn  10005    
#define PI 3.1415926  
#define lowbit(x) (x&-x)     
#define eps 1e-9
int a[maxn], b[maxn], id[maxn], ans[maxn], cnt;
void dfs(int la, int ra, int lb, int rb)
{
	if (la == ra)
		return;
	la++, rb--;
	int p = id[a[la]];
	if (p == rb)//说明只有一颗子树
	{  
		cnt++;
		dfs(la, ra, lb, rb);
		return;
	}
	int len = p - lb + 1;
	dfs(la, la + len - 1, lb, lb + len - 1);
	dfs(la + len, ra, lb + len, rb);
}
int main(void)
{
	int n, i, j;
	scanf("%d", &n);
	for (i = 1;i <= n;i++)
		scanf("%d", &a[i]);
	for (i = 1;i <= n;i++)
		scanf("%d", &b[i]), id[b[i]] = i;
	dfs(1, n, 1, n);
	int len = 1;ans[1] = 1;
	while (cnt--)
	{
		for (i = 1;i <= len;i++)
			ans[i] *= 2;
		for (i = 1;i <= len;i++)
			ans[i + 1] += ans[i] / mod, ans[i] %= mod;
		if (ans[len + 1])
			len++;
	}
	printf("%d", ans[len]);
	for (i = len - 1;i > 0;i--)
		printf("%09d", ans[i]);
	printf("\n");
	return 0;
}