BZOJ 4726 Sabota? （树型dp）

4726: [POI2017]Sabota?

Time Limit: 20 Sec Memory Limit: 128 MBSec Special Judge

Description

某个公司有n个人, 上下级关系构成了一个有根树。其中有个人是叛徒(这个人不知道是谁)。对于一个人, 如果他
下属(直接或者间接, 不包括他自己)中叛徒占的比例超过x，那么这个人也会变成叛徒，并且他的所有下属都会变
成叛徒。你要求出一个最小的x，使得最坏情况下，叛徒的个数不会超过k。

Input

第一行包含两个正整数n,k(1<=k<=n<=500000)。
接下来n-1行，第i行包含一个正整数p[i+1]，表示i+1的父亲是pi+1。

Output

输出一行一个实数x，误差在10^-6以内都被认为是正确的。

Sample Input

9 3
1
1
2
2
2
3
7
3

Sample Output

0.6666666667

HINT

答案中的x实际上是一个无限趋近于2/3但是小于2/3的数
因为当x取2/3时，最坏情况下3，7，8，9都是叛徒，超过了k=3。

Source

鸣谢Claris上传

思路：
由于题目的性质：
首先显然最坏情况初始的叛徒肯定是叶子
叛徒只有一个，不会同时出现两个没有仍和关系的子树都被感染为叛徒传染，并且带头叛变的人一定是从某个叶子往上走一条链，因为如果i没有带头叛变，那么i的父亲也一定不会带头叛变
f[i]表示i不带头叛变的话最小的x
那么我们对所有子树大小>k的f值取max即是答案
每个节点的状态？
以该节点为根节点的子树全 [变为叛徒 / 没有全变为叛徒]
组成该状态的情况？
[没有全变为叛徒]：
【1】某一个子树全是叛徒，但是在容许比例之内
【2】某个子树中有叛徒，但是其容忍度导致该子树没全变为叛徒
[全变为叛徒]：
自己的某一个子树中有叛徒，且叛徒比例大于容许比例
f[i]=max (min(f[j],siz[j]/(siz[i]-1)) （j为i的儿子）
因为对于i的一个儿子j，假如i因为j的子树里的叛徒比例大于x而带头叛变，那么既要满足x<=(j的子树大小占i的子树大小的比例)，还要满足j带头叛变即x<=f[j]，所以对两个量取min
那么如果i不叛变，那么就不能满足任意一个条件，所以对所有的取max
对于叶子，f[i]=1，因为不管怎样叶子本身就是叛徒，可以视为不需要条件就可以带头叛变，即只有当x>1时才不会叛变

#include <iostream>  
#include <cstdio>  
#include <cstring>  
#include <cmath>  
#include <algorithm>  
#define N 500010  
#define LL long long  
using namespace std; 

struct Edge{  
    int to;  
    int nxt;  
}ed[N];  

int n, k;  
int head[N], idc;  
double f[N], ans=0;  
int siz[N];  

inline void adde(int u, int v){  
    ed[++idc].to = v;  
    ed[idc].nxt = head[u];  
    head[u] = idc;  
}

void dfs(int u){  
    siz[u] = 1;  
    for(int i=head[u]; i; i=ed[i].nxt){  
        int v = ed[i].to;  
        dfs( v );  
        siz[u] += siz[v];  
    }  
    if( !head[u] ){//叶子节点x最大为1
        f[u] = 1;
    }  
    if( head[u] ){  
        for(int i=head[u]; i; i=ed[i].nxt){  
            int v = ed[i].to;
            f[u] = max(f[u], min(1.0 * siz[v] / (siz[u]-1), f[v]));
            //1.一个子树全为叛徒（不包括自己）
            //2.父亲的f大于等于儿子的f
            //因为是最坏情况，所以叛徒的起点可以在任意子树中，外层取max
            //而让不让子树被填满是我们的选择，所以内层取min 
        }  
    }  
    if(siz[u] > k){  
        ans = max(ans, f[u]);  
    }  
}

int main(){  
    scanf("%d%d", &n, &k);  
    for(int i=2; i<=n; i++){  
        int x; scanf("%d", &x);  
        adde(x, i);  
    }  
    dfs( 1 );  
    printf("%.10lf\n", ans);  
    return 0;  
}