牛客算法周练2 D题 树建立 思维

**D. 小H和游戏**

链接:(https://ac.nowcoder.com/acm/contest/5203/D)

题目描述

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

#include<bits/stdc++.h>
#define ll long long

using namespace std;

int fa[750001];
vector<int>G;
int fath[75000][3]; // 0代表自己受伤的次数
// 1代表它的儿子被波及的次数    //2代表自己孙子被波及的次数
void dfs(int u,int f){   //建树

    fa[u]=f;
    for (int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(f==v) continue;   //防止出现 fa[2]=1 fa[1]=2  建树
        dfs(v,u);
    }

}
/*
1 2 
2 3 
3 4 建成的树
      0
      |
      1
     / \
    3   2
   / \  
  4   5
  
*/
int main()
{
    int n,q;
    cin>>n>>q;
    for (int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        G[a].push_back(b);
        G[b].push_back(a);
    }
    dfs(1,0); //将最大的根节点的父亲设为0
    while(q--){
        ll ans=0;
        int m;
        cin>>m;
        fath[m][1]++;  //伤害波及到它的儿子
        fath[m][2]++;   //伤害波及到它的孙子
        fath[fath[m]][0]++; //伤害波及到它的父亲
        fath[fath[m]][1]++; //伤害波及它父亲的儿子(包括自己)   
        //就是父亲它的儿子被其他儿子波及以及父亲自己被伤害时波及它儿子总次数  
        //也就是它的每个儿子 所受伤的总次
        fath[fath[fath[m]]][0]++; //伤害波及到它的爷爷
        ans=fath[m][0]+fath[fath[m]][1]+fath[fath[fath[m]]][2];
        //fath[m][0]是由它的孙子波及的以及儿子波及的 
        //fath[fath[m]][1] 是它的父亲伤害的次数以及它父亲其它儿子波及到它总数
        //fath[fath[fath[m]]][2] 它的爷爷自己被攻击次数 (不包括爷爷被)
        cout<<ans;
    }
}