这道题目可以用树上差分来实现,由于以 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;
}