NC 15158. 小H和游戏

Link
题意：
树形图上，对某点进行轰炸，会波及到距离该点不超过 \(2\) 的点
\(q\) 次轰炸，每次都要输出该点受损次数
思路：
记 \(tot[x][0]\) 为 \(x\) 受损的次数，\(tot[x][1]\) 为 \(x\) 的孩子受损的次数，\(tot[x][2]\) 的孩子的孩子受损的次数
假如轰炸 \(x\)：
\(x\) 的父亲的父亲受损：\(tot[fa[fa[x]]][0]++\)
\(x\) 的父亲受损：\(tot[fa[x]][0]++\)
\(x\) 以及 \(x\) 的兄弟受损：\(tot[fa[x]][1]++\)
\(x\) 的孩子受损：\(tot[x][1]++\)
\(x\) 的孩子的孩子受损：\(tot[x][2]++\)
\(x\) 的总受损次数为：\(tot[x][0]+tot[fa[x]][1]+tot[fa[fa[x]]][2]\)
代码：

#include<bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
typedef long long ll;
typedef pair<int,int> pii;
//head
const int N=750005;
vector<int>g[N];
int fa[N],tot[N][3];
void dfs(int u) {
    for(auto v:g[u]) {
        if(v==fa[u]) continue;
        fa[v]=u;
        dfs(v);    
    }
}
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,q;
    cin>>n>>q;
    for(int i=1;i<n;i++) {
        int u,v;
        cin>>u>>v;
        g[u].pb(v);
        g[v].pb(u);
    }
    dfs(1);
    for(int i=1;i<=q;i++) {
    	int x;
    	cin>>x;
    	tot[x][1]++,tot[x][2]++;
    	if(fa[fa[x]]) tot[fa[fa[x]]][0]++;
    	if(fa[x]) tot[fa[x]][0]++,tot[fa[x]][1]++;
    	else tot[x][0]++;
    	cout<<tot[x][0]+tot[fa[x]][1]+tot[fa[fa[x]]][2]<<endl;
    }
    return 0;
}