悬赏关注，MnZn学淀粉质10^-114514秒求助只过#8其他万紫千红

#include<bits/stdc++.h>
#define N 10005
using namespace std;
int n,m,q[105],sz[N],sum,rt,mx[N],d[N],cnt,dd[N];
vector<pair<int,int>>g[N];
queue<int>p;
bool ok[10000005],del[N],ans[105];
void gravity(int u,int fa){
    sz[u]=1;
    mx[u]=0;
    for(auto[v,w]:g[u]){
        if((v^fa)&&!del[v]){
            gravity(v,u);
            mx[u]=max(mx[u],sz[v]);
            sz[u]+=sz[v];
        }
    }
    mx[u]=max(mx[u],sum-sz[u]);
    if(mx[u]<mx[rt]){
        rt=u;
    }
}
void dis(int u,int fa){
    dd[++cnt]=d[u];
    for(auto[v,w]:g[u]){
        if((v^fa)&&!del[v]){
            d[v]=d[u]+w;
            dis(v,u);
        }
    }
}
void treedivide(int u,int fa){
    ok[0]=del[u]=1;
    p.push(0);
    for(auto[v,w]:g[u]){
        if((v^fa)&&!del[v]){
            d[v]=w;
            dis(v,u);
            for(int i=1;i<=cnt;++i){
                for(int j=1;j<=m;++j){
                    if(q[j]>=dd[i]){
                        ans[j]|=ok[q[j]-dd[i]];
                    }
                }
            }
            for(int i=1;i<=cnt;++i){
                if(dd[i]<=1e7){
                    p.push(dd[i]);
                    ok[dd[i]]=1;
                }
            }
            cnt=0;
        }
    }
    while(p.size()){
        ok[dd[p.front()]]=0;
        p.pop();
    }
    for(auto[v,w]:g[u]){
        if((v^fa)&&!del[v]){
            sum=sz[v];
            mx[rt=0]=1e9;
            gravity(v,u);
            treedivide(rt,0);
        }
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1,u,v,w;i<n;++i){
        scanf("%d%d%d",&u,&v,&w);
        g[u].emplace_back(v,w);
        g[v].emplace_back(u,w);
    }
    sum=n;
    mx[0]=1e9;
    gravity(1,0);
    for(int i=1;i<=m;++i){
        scanf("%d",&q[i]);
    }
    treedivide(rt,0);
    for(int i=1;i<=m;++i){
        if(ans[i]){
            puts("AYE");
        }else{
            puts("NAY");
        }
    }
}