[BZOJ]4726 [POI2017] Sabota? 树形Dp

4726: [POI2017]Sabota?

Time Limit: 20 Sec   Memory Limit: 128 MBSec   Special Judge
Submit: 646   Solved: 272
[ 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上传

[ Submit][ Status][ Discuss] 

HOME Back

f[u]表示u这个子树未被感染完的最小x. 所以f[u] = max(f[u], min(f[v], 1.0 * siz[v] / (siz[u] - 1.0)));有的时候v整棵子树被感染但是放眼整棵u子树内， 这个比例是很小的. 因为要最坏情况， 所以要取max. 答案是对所有siz > k的点的f值取个max. 因为虽然f定义如此但是不能保证子树里被感染不超过k， 所以答案不是f[1]. 所以要保证所有siz>k的都不被感染. 中间取max即可保证 -- 因为计算了所有情况.

#include<stdio.h>
#include<algorithm>
using namespace std;
const int maxn = 500010;
double f[maxn], ans;
int n, K, num, h[maxn], siz[maxn];
struct edge{ int nxt, v;}e[maxn];
inline void add(int u, int v){e[++num].v = v, e[num].nxt = h[u], h[u] = num;}
void dfs(int u){
	siz[u] = 1;
	for(int i = h[u]; i; i = e[i].nxt)
		dfs(e[i].v), siz[u] += siz[e[i].v];
	if(siz[u] == 1) f[u] = 1;
	for(int i = h[u]; i; i = e[i].nxt)
		f[u] = max(f[u], min(f[e[i].v], 1.0 * siz[e[i].v] / (siz[u] - 1.0)));
	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);
		add(x, i);
	}
	dfs(1);
	printf("%.10f\n", ans);
}