CF #781 - C. Tree Infection（贪心，二分）

https://codeforces.com/contest/1665/problem/C

题意
给定一棵树，根节点为 1。起初每个节点都是健康的。

每一秒钟都可以进行两种操作：

• 扩散：对于每个节点 v，如果存在子节点被感染，那么可以再选择节点 v 的一个子节点将其感染；
• 指定感染：指定一个节点将其感染。

问，将所有节点都感染最少花费多少时间？

$2\le n\le 2\cdot 10^5$

思路
一开始题目读假了啊，没看到第一种操作中的每个节点。。
wa 了之后看了半个多钟头没看出来哪错了。。

一个点感染了可以扩散到兄弟节点，一堆兄弟节点之间没有一个感染的，那么这些节点都不会被扩散。
所以要先将其中的一个用操作 2 指定感染，假设互不相交的节点块一共有 x 个，那么就要至少花费 x 秒。

如果按一开始错误的思路，每次操作 1 只作用于一个节点，那么就是先把每个块中拿出来一个用于指定感染，剩下的就放一起，一共 sum 个。除了第一秒，每次指定感染都能扩散感染一个节点，所以指定感染 x 次的话，就能扩散感染 x-1 个，sum 减去 x-1，然后对于剩下没感染的节点，每一秒钟可以感染两个，花费除 2 上取整秒。

而现在每次操作 1 都可以作用于每个节点块，也就是说，假设一共有 x 个节点块，对于第 i 个进行指定感染的节点块，在 x 个节点块都被指定感染过之后，可以扩散感染 x - i 次。
而 x 个节点块是肯定要指定感染的，就相当于第 i 个进行指定感染的节点块可以免费扩散感染 x-i 个节点。
所以肯定是把大的节点块先进行指定感染，把节点块中剩下的节点尽可能多的免费感染。

在所有的节点块都被指定感染，免费扩散感染之后，如果还剩下一些节点没有被感染，那么就需要另外花费时间去感染他们。
多花费1秒，可以让所有的节点块都感染一个节点，然后再指定感染一个节点。
因为还要指定感染一个，所以不好直接求最少要另外花费多少秒，可以二分额外花费了 mid 秒，那么所有的节点块中剩余节点数都减去 mid，再把剩下的求和。看剩下的节点个数是否大于指定感染次数 mid。

#include<bits/stdc++.h>
using namespace std;

#define Ios ios::sync_with_stdio(false),cin.tie(0)

const int N = 200010, mod = 1e9+7;
int T, n, m;
int a[N];
int cnt[N];
int idx;

bool check(int mid)
{
	int sum = 0;
	for(int i=1;i<=idx;i++)
	{
		int x = a[i] - mid;
		if(x > 0) sum += x;
	}
	if(sum > mid) return 0;
	return 1;
}

signed main(){
	Ios;
	cin >> T;
	while(T--)
	{
		cin >> n;
		for(int i=1;i<=n;i++) cnt[i] = 0;
		
		for(int i=2;i<=n;i++)
		{
			int x; cin >> x;
			cnt[i] ++, cnt[x] ++;
		}
		
		idx = 0;
		a[++idx] = 0;
		a[++idx] = cnt[1] - 1;
		
		for(int i=2;i<=n;i++)
		{
			if(cnt[i] == 1) continue;
			a[++idx] = cnt[i] - 2;
		}
		sort(a+1, a+idx+1, greater<int>()); //把所有节点块从大到小排序
		
		int sum = 0;
		for(int i=1;i<=idx;i++)
		{
			a[i] -= idx-i;
		}
		int l = 0, r = 1e9;
		while(l < r)
		{
			int mid = l + r >> 1;
			if(check(mid)) r = mid;
			else l = mid + 1;
		}
		cout << idx + l << endl;
	}
	return 0;
}

仔细读题啊！
读假题害人啊！