loj#2665. 「NOI2013」树的计数

• 题目链接
• 题解
• 代码

---

---

题目链接

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