2024 普及暑假第七轮第三题：损失统计

有一个国家，其内部城市链接恰好是棵树，每年夏季，城市都会发生水灾，当一个城市发生水灾时，与最短距离不超过2的其他城市也发生水灾，你想知道，当水灾发生时，当前城市的受损次数。

输入

第1行，两个整数N(1≤N≤750000)、Q(1≤Q≤750000) 第2~N行，每行两个整数表示树上的一条边 第N+1~N+Q行，每行一个整数，表示发生水灾的城市

输出

输出Q行，每行一个整数表示这一次发生水灾城市在此次水灾后共计受损几次

样例

输入

4 4 
1 2 
2 3 
3 4 
1 
2 
3 
4

输出

1

2

3

3

提示

请结合样例推断

———————————————————————————————————————————

代码：

#include<bits/stdc++.h>
using namespace std;
int n,q,ans[750002];
vector<int> e[750002];
int pr(int x)
{
	for(int i=0;i<e[x].size();i++)
	{
		ans[e[x][i]]++;
		for(int j=0;j<e[e[x][i]].size();j++)
			if(e[e[x][i]][j]!=x) ans[e[e[x][i]][j]]++;
	}
	printf("%d\n",ans[x]);
}
int main()
{
	scanf("%d%d",&n,&q);
	for(int i=1;i<n;i++)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		e[a].push_back(b);
		e[b].push_back(a);
	}
	while(q--)
	{
		int x;
		scanf("%d",&x);
		ans[x]++;
		pr(x);
	}
	return 0;
}