E2. Escape The Maze (hard version) （思维 + 最短路）

Codeforces Round #756 (Div. 3)

Problem

有一个n个点的迷宫，迷宫有n - 1条边，是一棵树。A有k个朋友。A在节点1，k个朋友在k个不同的节点。每秒所有人都可以移动到相邻的一个节点中， 可以同时到达同一个节点。A和他的朋友可以在同一个节点或者在一条边的中点相遇。如果A在不遇到他的朋友的情况下，可以移动到任意一个叶子节点， A获胜，否则A的朋友们获胜。
问：如果A的朋友获胜，这k个人中最少几个人就可以抓A。

Solution

1. 首先考虑A如何获胜
   如果A到叶子节点的最小距离 小于 A到所有朋友的最小距离，那么A就获胜。反过来就是A的朋友们获胜。

2. A的朋友获胜，求解最小朋友数量

   1. 假设
      朋友节点：朋友所在的节点为朋友节点
      一个节点x有解：A如果通过x节点，A就会被他的朋友们抓住
      一个节点x无解：A可以通过x节点到达某个叶子节点
   2. 有解：
      ①x的节点中有一个朋友节点到x的距离 小于等于 A到x的距离，
      此时解为：1
      （因为A的这个朋友就可以提前到x节点，然后抓住A）
      ②x的节点中所有朋友节点到x的距离 大于 A到x的距离，且x的所有子节点有解，
      此时解为：x的所有子节点的解求和
      （若x的一个子节点无解，那么A就可以通过该节点获胜。也就是无解的情况）
   3. 无解：
      x的节点中所有朋友节点到x的距离 大于 A到x的距离，且其子节点中存在无解的节点。
      或者 x为非朋友节点的叶子节点（若x既是朋友节点，也是叶子节点，显然x节点有解）

3. 总之，这题很绕。。。需要耐心反复琢磨

Code

const int N = 2e5 + 5, M = 4e6 + 7;
vector<int> v[N];
int a[N], d[N], near[N];
int n, k;

int dfs(int x, int fa)
{
	int sum = 0; bool flag = 1;//子树和、是否有无解的子树
	for (int y : v[x])
	{
		if (y == fa) continue;
		d[y] = d[x] + 1;
		int c = dfs(y, x);
		if (c < 0) flag = 0;
		else sum += c;
		near[x] = min(near[x], near[y] + 1);
	}
	if (near[x] <= d[x]) return 1;//只要有一个朋友节点可以先到x就有解，哪怕其中有子节点无解
	if (sum == 0 || flag == 0) return -1;//x没有子节点且x不是朋友节点  或者  有一个子节点无解，则x节点无解
	return sum;
}

int main()
{
	IOS;
	int T; cin >> T;
	while (T--)
	{
		cin >> n >> k;
		for (int i = 1; i <= n; i++)
		{
			v[i].clear();
			near[i] = inf;
			d[i] = 0;
		}
		for (int i = 1; i <= k; i++)
		{
			cin >> a[i];
			near[a[i]] = 0;
		}
		for (int i = 1, x, y; i < n; i++)
		{
			cin >> x >> y;
			v[x].push_back(y);
			v[y].push_back(x);
		}
		
		cout << dfs(1, -1) << endl;
	}


	return 0;
}