bzoj4726 [POI2017]Sabota? 树形dp

Description

---

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

n,k(1<=k<=n<=500000)。
误差在10^-6以内都被认为是正确的。

Solution

---

一开始想着二分去了
正解是非常巧妙的树形dp做法，设f[i]表示以i为根的子树i不造反的最小比例，那么有f[i]=max(min(f[j],size[j]/(size[i]-1)))
因为只有一个叛徒，那么枚举叛徒所在的子树，贡献就很显然了。这里size[i]-1其实是为了保证答案非常接近但是不相等
初始状态f[叶子]=1，所有size>k的节点取max

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

const int N=800005;
const int E=800005;

struct edge {int y,next;} e[E];

int ls[N],size[N],edCnt;
double f[N];

void add_edge(int x,int y) {
    e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
}

void dfs(int x) {
    size[x]=1;
    for (int i=ls[x];i;i=e[i].next) {
        dfs(e[i].y);
        size[x]+=size[e[i].y];
    }
    f[x]=(size[x]==1);
}

void dp(int x) {
    for (int i=ls[x];i;i=e[i].next) {
        dp(e[i].y);
        f[x]=std:: max(f[x],std:: min(f[e[i].y],(double)size[e[i].y]/(size[x]-1.0)));
    }
}

int main(void) {
    int n,m; scanf("%d%d",&n,&m);
    rep(i,2,n) {
        int fa; scanf("%d",&fa);
        add_edge(fa,i);
    }
    dfs(1); dp(1);
    double ans=0;
    rep(i,1,n) if (size[i]>m) ans=std:: max(ans,f[i]);
    printf("%.10lf\n", ans);
    return 0;
}