[NOI2013]树的计数

题目描述

能评测了哦。

我们知道一棵有根树可以进行深度优先遍历（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

输入输出格式

输入格式：

 

输入文件count.in共有3行。

第一行包含1个正整数n，表示树的节点个数。

第二行包含n个正整数，是一个1~n的排列，表示树的DFS序。

第三行包含n个正整数，是一个1~n的排列，表示树的BFS序。

输入保证至少存在一棵树符合给定的两个序列。

 

输出格式：

 

输出文件count.out仅包含1个实数，四舍五入保留恰好三位小数，表示树高的平均值。

 

输入输出样例

输入样例#1：

5 
1 2 4 5 3 
1 2 3 4 5

输出样例#1：

3.500

说明

【评分方式】

如果输出文件的答案与标准输出的差不超过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的后方。

【时间限制】

1s 【空间限制】

256000KB

这是一道思维好题.

为了方便考虑,先安装bfs序重新编号,这样bfn[i]=i.

若两个点之间必须分层,那么对答案的贡献就是1,可分可不分对答案的贡献为0.5.

先从bfs序考虑.

若dfn[i]<dfn[i-1]这时i和i-1必须分层.

再考虑dfs序中的相邻的两个点,x,y且bfn[x]<bfn[y].

这时x和y就最多分一层,就是要么y是x的儿子,要么y是x的右边的兄弟.

那么在bfn[x]-bfn[y]之间就最多分一层.

设s[i]代表i与i-1之间是否分层.

这样就会有一些不等关系.

那么先把必须分层记录一下,然后在不等关系里面判断一下是否分了,若已经分了的话,那么就把这一段区间标记为强制不分层.

最后那些没有标记的点对答案的贡献为0.5

#include<bits/stdc++.h>
#define maxn 200010
using namespace std;
int pos[maxn],dfn[maxn],s[maxn],c[maxn];
double ans=1.0;
int main(){
  int n,x;
  scanf("%d",&n);
  for(int i=1;i<=n;i++)
    scanf("%d",&dfn[i]),pos[dfn[i]]=i;
  for(int i=1;i<=n;i++)
    scanf("%d",&x),dfn[pos[x]]=i;
  for(int i=1;i<=n;i++) pos[dfn[i]]=i;
  s[2]=1;
  for(int i=3;i<=n;i++)
    if(pos[i]<pos[i-1])s[i]++,ans+=1.0;
  for(int i=2;i<=n;i++) s[i]+=s[i-1];
  for(int i=3;i<=n;i++)
    if(dfn[i-1]<dfn[i] && s[dfn[i]]-s[dfn[i-1]])
      c[dfn[i-1]+1]++,c[dfn[i]+1]--;
  for(int i=3;i<=n;i++){
    c[i]+=c[i-1];
    if(!c[i]) ans+=0.5;
  }
  printf("%.3lf",ans+1.0);
  return 0;
}