BZOJ 4726 POI 2017 Sabota? 树形DP

4726: [POI2017]Sabota?

Time Limit: 20 Sec  Memory Limit: 128 MBSec  Special Judge
Submit: 128  Solved: 49
[Submit][Status][Discuss]

Description

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

Input

第一行包含两个正整数n,k(1<=k<=n<=500000)。

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

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上传

Solution

考虑不满足条件的情况，肯定是叶节点中出现了叛徒，然后不断往上传染。

考虑树形DP，f[i]为i节点不变成叛徒的最小的比例，当i为叶节点时，f[i] = 1.0。

转移：f[i] = max{f[i], min(f[son[i]], 1.0*siz[son[i]]/(son[i]-1))};

其儿子不变成叛徒的最小比例为x，小于x，则其儿子会变成叛徒；其儿子所占的比例为y，小于y，则其儿子变成叛徒后会影响到它。

故取p = min(x,y)，小于p，则它会变成叛徒。取max{p}，是为了让它的所有儿子都不影响到它。

答案就是所有子树大小大于k的f[i]的max值。

且需要注意，一开始只有一个叛徒，也就是说传染最多只会传染完某一棵子树。

Code

 #include <bits/stdc++.h>

 using namespace std;

 const int maxn = 5e5+;
 const double EPS = 1e-;
 #define REP(i, a, b) for (int i = (a), i##_end_ = (b); i <= i##_end_; ++i)
 #define REP_EDGE(i, a) for (int i = (a); i != -1; i = e[i].nxt)
 #define mset(a, b) memset(a, b, sizeof(a))
 int fcmp(double x) { if (x > -EPS && x < EPS) return ; return x < -EPS ? - : ; }
 double max_(const double &AI, const double &BI) { return fcmp(AI-BI) ==  ? AI : BI; }
 double min_(const double &AI, const double &BI) { return fcmp(AI-BI) == - ? AI : BI; }
 int n, k;
 struct Edge
 {
     int v, nxt;
     Edge (int v = , int nxt = ): v(v), nxt(nxt) {}
 }e[maxn];
 int head[maxn], label, siz[maxn];
 double f[maxn];

 void ins(int u, int v) { e[++label] = Edge(v, head[u]), head[u] = label; }

 void dfs(int u, int fa)
 {
     siz[u] = ;
     REP_EDGE(i, head[u])
         if (e[i].v != fa)
             dfs(e[i].v, u), siz[u] += siz[e[i].v];
     if (siz[u] == ) { f[u] = 1.0; return ; }
     f[u] = 0.0;
     REP_EDGE(i, head[u])
         if (e[i].v != fa)
             f[u] = max_(f[u], min_(1.0*siz[e[i].v]/double(siz[u]-), f[e[i].v]));
 }

 int main()
 {
     scanf("%d %d", &n, &k);
     REP(i, , n) head[i] = -;
     label = -;
     REP(i, , n){ int u; scanf("%d", &u), ins(u, i); }
     dfs(, );
     double ans = 0.0;
     REP(i, , n)
         if (siz[i] > k) ans = max_(ans, f[i]);
     printf("%.8lf\n", ans);
     return ;
 }