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  
#include         
#include                
#include  
#include        
#include      
#include 
#include  
#include                
#include                
#include                
#include        
#include       
#include        
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;
}


你可能感兴趣的:(搜索)