loj#2665. 「NOI2013」树的计数

---

目录

• [TOC]

题目链接

loj#2665. 「NOI2013」树的计数

题解

求树高的期望

对bfs序分层

考虑同时符合dfs和bfs序的树满足什么条件

第一个点要强制分层

对于bfs序连续的a,b两点,若a的bfs序小于b的bfs序,且a的dfs序大于b的,那么它们之间肯定要分层,对答案贡献为1

对于dfs序连续的a,b两点,若a的dfs序小于b的,且a的bfs序也小于b,那么它们的深度差不超过1，也就是说它们在的bfs序上之间最多分一层

先把前两个条件都判一下，然后把第2个条件判一下(如果它们之间已经分层了，那么就强制其他的不分层)

最后剩下的可分层可不分的点，贡献是0.5

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define gc getchar()
#define pc putchar
inline int read() {
	int x = 0,f = 1;
	char c = getchar();
	while(c < '0' || c > '9') c = gc;
	while(c <= '9' && c >= '0') x = x * 10 + c - '0',c = gc;
	return x * f;
}
void print(int x) {
 	if(x < 0) {
 		pc('-');
 		x = -x;
 	}
 	if(x >= 10) print(x / 10);
 	pc(x % 10 + '0');
 }
const int maxn = 200007;
 int n,top = 0,q[maxn];
 double ans = 0;
 int P[maxn],D[maxn],B[maxn];
 int x[maxn],y[maxn];
inline void mark(int a,int b) {
	++ y[a], -- y[b];
}
 int main() {
 	n = read();
 	for(int j,i = 1;i <= n;++ i) P[j = read()] = i;  //dfs中第j个点排在第i位
 	for(int j,i = 1;i <= n;++ i) D[B[i] = P[j = read()]] = i;
 	//B:bfs序中排在第i的点在dfs序中排B[i]
 	//D:dfs序中排在第i的点在bfs序中排D[i]
 	for(int i = 1;i < n;++ i) {
 		x[i] = x[i - 1];
 		if(i == 1 || B[i] > B[i + 1]) //lev[i] < lev[i + 1]
 			ans += 2.0,
 			++ x[i],
 			mark(i,i + 1);
 	}
 	for(int i = 1;i < n;++ i) {
 		if(D[i] < D[i + 1]) {
 			if(x[D[i + 1] - 1] > x[D[i] - 1]) //层数小
 					mark(D[i],D[i + 1]);
 			else q[++ top] = D[i]; //
 		}
	}
	for(int i = 1;i <= n;++ i) y[i] = y[i] + y[i - 1];
 	for(int i = 1;i <= top;++ i) ans += (y[q[i]] == 0);
 	printf("%.3lf\n%.3lf\n%.3lf\n",ans / 2 + 1 - 0.001,ans / 2 + 1,ans / 2 + 1 + 0.001);
 	return 0;
 }