牛客-苹方树

题目链接：苹方树

这道题目可以用树上差分来实现，由于以 i 为根的子树如果包含了第 j 条路径中至少一个节点，那么 vi 就要乘上 lj，假设起点为x，终点为y，那么他们的祖先c = lca(x,y)，由题意得祖先只需除去一次即可，因为祖先以上的点都要乘w，那么再考虑数据范围，肯定会爆，那么因此我们从完全平方数的性质去考虑，所有的质因子为偶数，该数一定是一个完全平方数，那么因此我们可以对权值开二维空间，记录每个点的质因子的数量，最后判断是否都是偶数即可。

代码：

#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second
using namespace std;
const int inf = 0x3f3f3f3f3f3f3f3f;
const int N = 2e5+5;
vector<int>g[N];
int fa[N][35],dep[N];
int d[N][105];
int n,m,q;

void dfs(int x,int p){
    dep[x] = dep[p]+1;
    fa[x][0] = p;

    for(int i=1;i<=31;i++){
        fa[x][i] = fa[fa[x][i-1]][i-1];
    }

    for(const auto &y:g[x]){
        if(y==p)continue;
        dfs(y,x);
    }
    return;

}

int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);

    for(int i=31;i>=0;i--){
        if(dep[fa[x][i]]>=dep[y]){
            x=fa[x][i];
        }
    }

    if(x==y)return x;

    for(int i=31;i>=0;i--){
        if(fa[x][i]!=fa[y][i]){
            x=fa[x][i];
            y=fa[y][i];
        }
    }

    return fa[x][0];

}

void dfs1(int x,int p){
    for(const auto &y:g[x]){
        if(y==p)continue;
        dfs1(y,x);
        for(int i=2;i<=100;i++){
            d[x][i] += d[y][i];
        }
    }

    return;

}

void solve(){
    cin>>n;
    for(int i=1;i<n;i++){
        int x,y;cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1,0);

    cin>>m;
    for(int i=1;i<=m;i++){
        int x,y,w;cin>>x>>y>>w;
        int c = lca(x,y);
        for(int j=2;j*j<=w;j++){
            if(w%j==0){
                int res = 0;
                while(w%j==0){
                    w/=j;
                    res++;
                }
                d[x][j]+=res;
                d[y][j]+=res;
                d[c][j]-=res;
            }
        }
        if(w>1){
            d[x][w]++;
            d[y][w]++;
            d[c][w]--;
        }
    }

    dfs1(1,0);

    cin>>q;
    while(q--){
        int x;cin>>x;
        bool flag = true;
        for(int i=2;i<=100;i++){
            if(d[x][i]%2!=0)flag=false;
        }
        if(flag)cout<<"YES"<<"\n";
        else cout<<"NO"<<"\n";
    }

    return;

}

signed main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int t=1;
    while(t--){
        solve();
    }

    return 0;
}