Wannafly挑战赛10-C-小H和游戏（瞎搞）

题目描述：
小H正在玩一个战略类游戏，她可以操纵己方的飞机对敌国的N座城市(编号为1~N)进行轰炸
    敌国的城市形成了一棵树，小H会依次进行Q次轰炸，每次会选择一个城市A进行轰炸，和这座城市距离不超过2的城市都会受损(这里距离的定义是两点最短路径上的边数)，轰炸结束后，小H还想知道当前城市A受损的次数

    作为游戏的开发者之一，你有义务回答小H的问题

输入描述：
第1行，两个整数N(1≤N≤750000)、Q(1≤Q≤750000)
第2~N行，每行两个整数表示树上的一条边

第N+1~N+Q行，每行一个整数，表示小H这次轰炸的城市

输出描述：

输出Q行，每行一个整数表示这一次轰炸的城市在此次轰炸后共计受损几次

输入：

4 4 
1 2 
2 3 
3 4 
1 
2 
3 

4

输出：

1 
2 
3 

3

题解：先求一发每个点的父亲，然后设一个fb数组，记录距离该点为1和为2的儿子的受损情况，然后每次轰炸一个城市，则

fb[x][1]、fb[x][2]、fb[fa[x]][0]、fb[fa[x]][1]、fb[fa[fa[x]]][0]都加1，最后答案就是fb[x][0]+fb[fa[x]][1]+fb[fa[fa[x]]][2]

#include<vector>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxn 750005
int n,Q,fa[maxn],fb[maxn][3];
vector<int>q[maxn];
void dfs(int u,int p)
{
	fa[u]=p;
	for(int i=0;i<q[u].size();i++)
	{
		int v=q[u][i];
		if(v==p) 
			continue;
		dfs(v,u);
	}
}
int main(void)
{
	int x,y;
	scanf("%d%d",&n,&Q);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		q[x].push_back(y);
		q[y].push_back(x);
	}
	dfs(1,0);
	while(Q--)
	{
		scanf("%d",&x);
		fb[x][1]++;fb[x][2]++;
		fb[fa[x]][0]++;fb[fa[x]][1]++;
		fb[fa[fa[x]]][0]++;
		printf("%d\n",fb[x][0]+fb[fa[x]][1]+fb[fa[fa[x]]][2]);
	}
	return 0;
}