Wannafly挑战赛10 C 小H和游戏【延迟标记+并查集】

链接：https://www.nowcoder.com/acm/contest/72/C
来源：牛客网
题目描述
小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

分析：
一开始看成图了，直接暴力无限MLE+TLE。树操作，只有x节点的上下两层关系(表示方法不唯一);
G[i][1]；表示i节点的儿子；
G[i][2]；表示i节点的儿子和孙子；
F[i]：表示i节点被直接炸的次数；
查询时：(G[fa[x]][1] || F[x]) + G[x][2] + F[fa[x]] + F[fa[fa[x]]]；
更新时：F[x]++，G[fa[x]][1]++, G[fa[x]][2]++，G[fa[fa[x]]][2]++;
注意节点的父亲有无的情况。。。

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 750000 + 5;
int fa[MAXN], F[MAXN], G[MAXN][3];

int main() {
    int n, m, a, b;
    scanf("%d %d", &n, &m);
    for(int i = 1; i < n; ++i) {
        scanf("%d %d", &a, &b);
        if(!fa[a]) fa[a] = b; //并查集维护这棵树
        else fa[b] = a;
    }
   while(m--) {
        int x;
        scanf("%d", &x);
        int ans = 0;
        F[x]++;
        if(fa[x]) {
            G[fa[x]][1]++, G[fa[x]][2]++;
            ans = G[fa[x]][1];
            if(fa[fa[x]]) {
                G[fa[fa[x]]][2]++;
            }
        }
        else ans = F[x];
        printf("%d\n", ans + G[x][2] + F[fa[x]] + F[fa[fa[x]]]);
    }
    return 0;
}