BZOJ3244: [Noi2013]树的计数

本文介绍了一种计算给定DFS序和BFS序下所有可能的有根树的高度平均值的方法。通过分析DFS序和BFS序之间的关系,提出了一种有效算法来解决这个问题。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Description

我们知道一棵有根树可以进行深度优先遍历(DFS)以及广度优先遍历(BFS)来生成这棵树的DFS序以及BFS序。两棵不同的树的DFS序有可能相同,并且它们的BFS序也有可能相同,例如下面两棵树的DFS序都是1 2 4 5 3,BFS序都是1 2 3 4 5

现给定一个DFS序和BFS序,我们想要知道,符合条件的有根树中,树的高度的平均值。即,假如共有K棵不同的有根树具有这组DFS序和BFS序,且他们的高度分别是h1,h2,...,hk,那么请你输出
(h1+h2..+hk)/k

Input

有3行。 
第一行包含1个正整数n,表示树的节点个数。 
第二行包含n个正整数,是一个1~n的排列,表示树的DFS序。 
第三行包含n个正整数,是一个1~n的排列,表示树的BFS序。 
输入保证至少存在一棵树符合给定的两个序列。

Output

仅包含1个实数,四舍五入保留恰好三位小数,表示树高的平均值。

Sample Input


5
1 2 4 5 3
1 2 3 4 5

Sample Output

3.500

HINT

【评分方式】

如果输出文件的答案与标准输出的差不超过0.001,则将获得该测试点上的分数,否则不得分。

【数据规模和约定】



20%的测试数据,满足:n≤10;

40%的测试数据,满足:n≤100;

85%的测试数据,满足:n≤2000;

100%的测试数据,满足:2≤n≤200000。

【说明】

树的高度:一棵有根树如果只包含一个根节点,那么它的高度为1。否则,它的高度为根节点的所有子树的高度的最大值加1。

对于树中任意的三个节点a , b , c ,如果a, b都是c的儿子,则a, b在BFS序中和DFS序中的相对前后位置是一致的,即要么a都在b的前方,要么a都在b的后方。

Source

神题
首先把BFS变成1...n,然后对它按深度分层
设dfs序为a[i],pos[i]表示i在dfs序中的位置
第i层的深度为i,假设它包含[l,r],那么有以下限制
第1层是1
pos[l]<pos[l+1]<pos[l+2]...pos[r]
dep[a[i+1]]<=dep[a[i]]+1(因为a[i+1]只可能是a[i]或者a[i]祖先的儿子)
这样不好计算,我们把分层点提取出来,记为x
如果i和i+1在同一层,则x[i]=1否则x[i]=0
那么之前的限制变成了
x[1]=1
if(pos[i]>pos[i+1]) x[i]=1
if(a[i]<a[i+1]) ∑(j = a[i] to a[i+1]-1) x[j]<=1
同时答案为∑x[i]+1
考虑如何计算期望
第一个限制和第二个限制确定了一定x[i]为1的
对于第三个限制,如果某一段已经出现了x[i]=1,那么其他一定全0,打上差分标记
可以证明,剩下的第三个限制中,x[i]未确定的只有0或者1个,那么期望为0.5
#include <bits/stdc++.h>

using namespace std;

const int MAXN = 200020;

int a[MAXN], b[MAXN], pos[MAXN], x[MAXN], done[MAXN], s[MAXN], n, c[MAXN];

int main()
{
	scanf( "%d", &n );
	for( int i = 1 ; i <= n ; i++ ) scanf( "%d", &a[ i ] );
	for( int i = 1 ; i <= n ; i++ ) scanf( "%d", &b[ i ] ), c[ b[ i ] ] = i;
	for( int i = 1 ; i <= n ; i++ ) pos[ a[ i ] = c[ a[ i ] ] ] = i;
	x[ 1 ] = 1; done[ 1 ]++; done[ 2 ]--;
	for( int i = 1 ; i < n ; i++ ) if( pos[ i ] > pos[ i + 1 ] ) x[ i ] = 1, done[ i ]++, done[ i + 1 ]--;
	for( int i = 1 ; i <= n ; i++ ) s[ i ] = s[ i - 1 ] + x[ i ];
	for( int i = 1 ; i < n ; i++ ) if( a[ i ] < a[ i + 1 ] && s[ a[ i + 1 ] - 1 ] - s[ a[ i ] - 1 ] )
		done[ a[ i ] ]++, done[ a[ i + 1 ] ]--;
	double ans = 1;
	int c = 0;
	for( int i = 1 ; i < n ; i++ )
	{
		c += done[ i ];
		if( c ) ans += x[ i ];
		else ans += 0.5;
	}
	printf( "%.3lf\n%.3lf\n%.3lf\n", ans - 0.001, ans, ans + 0.001 );
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值