Wannaly挑战赛10-C小h和游戏

题目链接

C小h和游戏

---

题意

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

---

题解

cnt[]:表示直接炸到A城,cntf[]:表示炸到A的儿子而影响到A,cntff[]:表示炸到A的孙子而影响到A
于是炸到A时的更新为,cnt[a]++;cntf[fa[a]]++;cntff[fa[fa[a]]]++;
A点的答案为cnt[fa[a]] + cnt[fa[fa[a]]] + cntf[fa[a]] + cntf[a] + cntff[a]

---

代码

/**
    n个结点的树，现在进行Q次轰炸，轰炸其中的结点A,距离当前城市不超过2的城市都会受损，
    轰炸结束后，小H还想知道当前A城市受损的次数。

    cnt[]:表示直接炸到A城,cntf[]:表示炸到A的儿子而影响到A,cntff[]:表示炸到A的孙子而影响到A
    于是炸到A时的更新为,cnt[a]++;cntf[fa[a]]++;cntff[fa[fa[a]]]++;
    A点的答案为cnt[fa[a]] + cnt[fa[fa[a]]] + cntf[fa[a]] + cntf[a] + cntff[a]
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 8e5;
vector<int> way[maxn];
int fa[maxn],cnt[maxn],cntf[maxn],cntff[maxn];
void dfs(int x,int f)
{
    fa[x] = f;
    for(int i=0;i<way[x].size();i++) if(way[x][i]!=f) dfs(way[x][i],x);
}
int main()
{
    int n,q;
    while(~scanf("%d%d",&n,&q))
    {
        for(int i = 0;i < maxn ; i ++) way[i].clear();
        memset(cnt,0,sizeof(cnt));memset(cntf,0,sizeof(cntf));memset(cntff,0,sizeof(cntff));
        for(int i=1,u,v;i<n;i++)
        {
            scanf("%d%d",&u,&v);
            way[u].push_back(v);
            way[v].push_back(u);
        }
        dfs(1,0);
        while(q--)
        {
            int a;scanf("%d",&a);
            ++cnt[a];++cntf[fa[a]];++cntff[fa[fa[a]]];
            ll ans = cnt[fa[a]] + cnt[fa[fa[a]]] + cntf[fa[a]] + cntf[a] + cntff[a];
            printf("%lld\n",ans);
        }
    }
    return 0;
}