BZOJ3244 NOI2013树的计数(概率期望)

时间:2023-03-08 18:19:07

  容易发现的一点是如果确定了每一层有哪些点,树的形态就确定了。问题变为划分bfs序。

  考虑怎样划分是合法的。同一层的点在bfs序中出现顺序与dfs序中相同。对于dfs序中相邻两点依次设为x和y,y至多在x的下一层。特殊的,根单独作为一层。

  这些条件显然是必要的,考虑一种构造方案:对于某个点,如果其在dfs序中前一个点的下一层则直接连边,否则从其bfs序中前一个点(显然是与其在同一层的)的父亲连边。这看起来是没有问题的,所以这些条件也是充分的。

  接下来考虑怎么算答案。如果bfs序中相邻两点被划分在了不同层树高就会++,于是考虑相邻点是否能划在同一层。

  首先如果bfs序中相邻两点在bfs序和dfs序中的相对顺序不同,其必须划分,这保证了第一个条件满足。

  然后对dfs序中相邻两点x和y,其在bfs序中x到y(有序,若x在y后面则该限制无效)这段区间至多被划分一次,这显然保证了第二个条件满足。

  这样可以固定一些位置是否划分,保证了方案合法,划分一次提供1贡献。而对于剩下的位置不管是否划分都是合法的,所以提供0.5的贡献。

  至于具体怎么搞,第一个条件扫一遍即可确定。第二个条件若区间内已被划分打个标记即可。否则虽然看起来有点麻烦,但实际上如果出现这种情况该区间长度只会为1。因为这说明他们之间的dfs序和bfs序相对顺序都是相同的,这仅当两点在同一层才会出现,而这样他们在bfs序中也是相邻的。

  往往很多必要条件组合在一起就是充分的了,大力猜结论不要怂。

  bzoj上莫名其妙要输出三个数。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
int read()
{
int x=,f=;char c=getchar();
while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
return x*f;
}
#define N 200010
int n,a[N],b[N],id_dfs[N],id_bfs[N],f[N],s[N],cnt[N],ans=;
int main()
{
#ifndef ONLINE_JUDGE
freopen("bzoj3244.in","r",stdin);
freopen("bzoj3244.out","w",stdout);
const char LL[]="%I64d\n";
#else
const char LL[]="%lld\n";
#endif
n=read();
for (int i=;i<=n;i++) id_dfs[a[i]=read()]=i;
for (int i=;i<=n;i++) id_bfs[b[i]=read()]=i;
memset(f,,sizeof(f));f[]=;
for (int i=;i<n;i++)
if (id_dfs[b[i]]>id_dfs[b[i+]]) f[i]=;
for (int i=;i<n;i++) s[i]=s[i-]+(f[i]>);
for (int i=;i<n;i++)
if (id_bfs[a[i]]<id_bfs[a[i+]]&&s[id_bfs[a[i+]]-]>s[id_bfs[a[i]]-])
cnt[id_bfs[a[i]]]++,cnt[id_bfs[a[i+]]]--;
int x=;
for (int i=;i<n;i++)
{
x+=cnt[i];
if (x) f[i]=max(f[i],);
}
for (int i=;i<n;i++)
if (f[i]>=) ans+=f[i]<<;
else ans++;
//printf("%.3f",ans/2.0);
printf("%.3f\n%.3f\n%.3f",ans/2.0-0.001,ans/2.0,ans/2.0+0.001);
return ;
}