牛客算法周周练2 D小H和游戏 ( 树形结构 )

题目链接

解题报告：
对于被轰炸城市而言，它的孩子，孩子的孩子，父亲的父亲，父亲及父亲的孩子均受到了影响，暴力更改显然不行，不妨用标记更新，详细见代码。

#define first f
#define second s
#define ll long long
#define mp make_pair
#define pb push_back
#define pf push_front
#define lb lower_bound
#define ub upper_bound
#define sl(p) strlen(p)
#define SZ(p) p.size()
#define pii pair<int,int>
#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MOD=1e9+7;
const int maxn=750005;
const double eps=1e-9;
const int inf=0x3f3f3f3f;
const double PI=acos(-1.0);
const double e=2.718281828459;

struct node
{
    int to,next;
}edge[maxn<<1];
int head[maxn<<1],cnt,f[maxn];
int ans[maxn];//记录父亲或父亲的父亲的影响次数
int ant[maxn];//记录对孩子的影响次数
int knt[maxn];//距离对孩子的孩子的影响次数

void add(int from ,int to)
{
    edge[++cnt].next=head[from];
    edge[cnt].to=to;
    head[from]=cnt;
}
void dfs(int now,int pre)
{
    f[now]=pre;
    for(int i=head[now];~i;i=edge[i].next){
        int to=edge[i].to;
        if(to==pre) continue;
        dfs(to,now);
    }
}

int main()
{
    mem(head,-1);
    mem(ans,0);
    int n,q,u,v;
    scanf("%d%d",&n,&q);
    for(int i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    dfs(1,0);
    while(q--){
        scanf("%d",&u);
        ans[f[u]]++;ans[f[f[u]]]++;
        ant[f[u]]++;ant[u]++;
        knt[u]++;
        printf("%d\n",ans[u]+ant[f[u]]+knt[f[f[u]]]);
    }
    return 0;
}