bzoj 4726: [POI2017]Sabota? 树形dp

题意

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

分析

没想到啊2333

设f[i]表示i带头不叛变需要的最小x。
那么f[i]=max(min(f[j],size[j]/(size[x]-1)))且j是i的儿子。
为什么呢？
因为如果i因为j的叛变而叛变的话，首先x必须要小于f[j]，而且j的子树大小占i的子树大小必然要不大于x，所以要取min。因为i不能叛变，所以要取个max。
边界条件是f[叶节点]=1
答案就是所有size[x]>k的f[x]取个max就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=500005;

int n,m,size[N],last[N],cnt;
double f[N];
struct edge{int to,next;}e[N];

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void dp(int x)
{
    size[x]=1;
    for (int i=last[x];i;i=e[i].next) dp(e[i].to),size[x]+=size[e[i].to];
    if (!last[x]) f[x]=1;
    else for (int i=last[x];i;i=e[i].next) f[x]=max(f[x],min(f[e[i].to],1.0*size[e[i].to]/(size[x]-1)));
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=2;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        addedge(x,i);
    }
    dp(1);
    double ans=0;
    for (int i=1;i<=n;i++) if (size[i]>m) ans=max(ans,f[i]);
    printf("%.10lf",ans);
    return 0;
}