BZOJ 4726 [POI 2017] 树DP 解题报告

4726: [POI2017]Sabota?

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

【解题报告】

$dp[i]$表示无法使i的子树中所有人都变成叛徒的最小x，可以列出方程
$dp[i]=max( dp[i],min( dp[j],size[j]/( size[i]-1 ) ) )$ (j是i的儿子)
也就是说，如果$j$的子树大小所占比例高于$x$，并且让$j$全变成叛徒需要的$x$比当前值大，
都会使i也变成叛徒，所以$dp[i]$必须必他们的最小值大才可以
注意答案是所有满足$size[i]>k$的$dp[i]$的最大值

代码如下：

/**************************************************************
    Problem: 4726
    User: onepointo
    Language: C++
    Result: Accepted
    Time:9908 ms
    Memory:35908 kb
****************************************************************/

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 500010

int cnt,head[N];
struct Edge{int to,nxt;}e[N<<1];
int n,m,size[N];
double dp[N],ans;

void adde(int u,int v)
{
    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;
    e[++cnt].to=u;e[cnt].nxt=head[v];head[v]=cnt;
}
void dfs(int u,int fa)
{
    size[u]=1;
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dfs(v,u);
        size[u]+=size[v];
    }
    if(size[u]==1) dp[u]=1.0;
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        if(v==fa) continue;
        dp[u]=max(dp[u],min(dp[v],1.0*size[v]/(size[u]-1.0)));
    }
}
int main()
{
    cnt=-1;
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=2;i<=n;++i)
    {
        int a;scanf("%d",&a);
        adde(a,i);  
    }
    dfs(1,1);
    for(int i=1;i<=n;++i) 
        if(size[i]>m) ans=max(ans,dp[i]);
    printf("%.10f",ans);
    return 0;
}