牛客算法周周练2 - 小H和游戏(DFS)
牛客算法周周练2 - 小H和游戏(DFS)
链接:https://ac.nowcoder.com/acm/contest/5203/D
来源:牛客网
时间限制:C/C++ 2秒,其他语言4秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
小H正在玩一个战略类游戏,她可以操纵己方的飞机对敌国的N座城市(编号为1~N)进行轰炸
敌国的城市形成了一棵树,小H会依次进行Q次轰炸,每次会选择一个城市A进行轰炸,和这座城市距离不超过2的城市都会受损(这里距离的定义是两点最短路径上的边数)
轰炸结束后,小H还想知道当前城市A受损的次数
作为游戏的开发者之一,你有义务回答小H的问题
作为游戏的开发者之一,你有义务回答小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
对于城市u, 我们考虑用一个fa数组找到城市的祖先fa[u], 它的祖先的祖先就是fa[fa[u]]
num[i][0]表示i点对自身的影响,num[i][1]表示i点对儿子的影响,num[i][2]表示i点对孙子的影响
每次查询我们都直接更新num即可
轰炸x会对以下点造成贡献:
距离0:x本身
距离1:x的父亲和儿子
距离2:x的爷爷和孙子和兄弟
这里要特别注意的是很容易把num[fa[x]][1]++写成num[x][0]++。因为一个点被轰炸,它的兄弟结点肯定全被轰炸。
最后输出的是num[x][0]+num[fa[x]][1]+num[fa[fa[x]]][2]
1 #include <bits/stdc++.h> 2 typedef long long LL; 3 #define pb push_back 4 const int INF = 0x3f3f3f3f; 5 const double eps = 1e-8; 6 const int mod = 1e9+7; 7 const int maxn = 8e5+10; 8 using namespace std; 9 10 vector<int> vt[maxn]; 11 int fa[maxn]; 12 int num[maxn][3]; 13 14 void DFS(int u,int pre) 15 { 16 fa[u]=pre; 17 for(int i=0;i<vt[u].size();i++) 18 { 19 int v=vt[u][i]; 20 if(v!=pre) DFS(v,u); 21 } 22 } 23 24 int main() 25 { 26 #ifdef DEBUG 27 freopen("sample.txt","r",stdin); //freopen("data.out", "w", stdout); 28 #endif 29 30 int n,q; 31 scanf("%d %d",&n,&q); 32 for(int i=1;i<n;i++) 33 { 34 int u,v; 35 scanf("%d %d",&u,&v); 36 vt[u].pb(v); vt[v].pb(u); 37 } 38 DFS(1,0); 39 for(int i=1;i<=q;i++) 40 { 41 int x; 42 scanf("%d",&x); 43 num[fa[fa[x]]][0]++;//爷爷节点对自己影响+1 44 num[fa[x]][0]++;//父亲节点对自己影响+1 45 num[fa[x]][1]++;//父亲节点对儿子影响+1,注意不要写成num[x][0]++了 46 num[x][1]++;//x对儿子节点影响+1 47 num[x][2]++;//x对孙子节点影响+1 48 printf("%d\n",num[x][0]+num[fa[x]][1]+num[fa[fa[x]]][2]); 49 } 50 51 return 0; 52 }
-