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行，每行一个整数表示这一次轰炸的城市在此次轰炸后共计受损几次

示例1

输入

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

输出

1 
2 
3 
3

题意：字面意思
这个题 
这个题可以说是构图，也可以是构树，因为只有n-1条边。嘿嘿嘿。
解题思路：首先我想到的是建图（我最乐意做的事情了。），然后纯暴力模拟，结果卡90%，超时。
然后我就想到了并查集，但是涉及到路径压缩啥的，关系就没了呀。
想了十几分钟，还没想到。（有点蠢）。
然后我就对自己说，是时候学点新姿势了。
然后看了下别人的代码，秒懂。


开一个数组，这个数组的意义是对于当前结点 距离为1，距离为2的儿子和孙子造成影响的的次数
如果我对a进行轰炸，那我只需要在num[a][1],num[a][2],num[father[a]][1],num[father[father[a]]][0],加一次就行。我就能表示树上？？？ why?(自己思考)

然后每次输出num[a][0]+num[father[a]][1]+num[father[father[a]]][2]就行了（？）

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn=750000+50;
int head[maxn],n,m,tot,father[maxn];
int num[maxn][3];
struct Node{
	int v,next;
}node[maxn*3];
void add(int u,int v){
	node[tot].v=v;node[tot].next=head[u];
	head[u]=tot++;
}
void dfs(int u,int from){
	father[u]=from;
	for(int i=head[u];~i;i=node[i].next){
		int v=node[i].v;
		if(v!=from){
			dfs(v,u);
		}
	}
}
int main(){
	int i,j;
	scanf("%d%d",&n,&m);
	memset(num,0,sizeof(num));
	memset(head,-1,sizeof(head));
	int u,v;tot=0;
	for(i=2;i<=n;i++){
		scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
	}
	memset(num,0,sizeof(num));
	dfs(1,0);
	while(m--){
		scanf("%d",&u);
		num[u][1]++;num[u][2]++;
		num[father[u]][0]++;num[father[u]][1]++;
		num[father[father[u]]][0]++;
		printf("%d\n",num[u][0]+num[father[u]][1]+num[father[father[u]]][2]);
	}
	return 0;
}

新姿势（GET）