时间复杂度:
洛谷P3806(100分加强数据版)
#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define per(i,n,a) for(int i=n;i>=a;i--)
#define endl '\n'
#define mem(a,b) memset(a,b,sizeof(a))
#define IO ios::sync_with_stdio(false);cin.tie(0);
using namespace std;
const int INF=0x3f3f3f3f;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int mod=1e9+7;
const int maxn=1e5+5;
const int maxk=1e7+5;
const int limit=1e7;
int tot,head[maxn];
struct E{
int to,next,w;
}edge[maxn<<1];
void add(int u,int v,int w){
edge[tot].to=v;
edge[tot].w=w;
edge[tot].next=head[u];
head[u]=tot++;
}
int n,m,rt,sum,cnt,q[maxn];
int tmp[maxn],siz[maxn],dis[maxn],maxp[maxn];
bool judge[maxk],ans[maxn],vis[maxn];
void getrt(int u,int f){
siz[u]=1,maxp[u]=0;
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v==f||vis[v]) continue;
getrt(v,u);
siz[u]+=siz[v];
if(siz[v]>maxp[u]) maxp[u]=siz[v];
}
maxp[u]=max(maxp[u],sum-siz[u]);
if(maxp[u]<maxp[rt]) rt=u;
}
void getdis(int u,int f){
if(dis[u]<=limit) tmp[cnt++]=dis[u];
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v==f||vis[v]) continue;
dis[v]=dis[u]+edge[i].w;
getdis(v,u);
}
}
void solve(int u){
queue<int> que;
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(vis[v]) continue;
cnt=0;
dis[v]=edge[i].w;
getdis(v,u);
for(int j=0;j<cnt;j++)
for(int k=0;k<m;k++)
if(q[k]>=tmp[j])
ans[k]|=judge[q[k]-tmp[j]];
for(int j=0;j<cnt;j++){
que.push(tmp[j]);
judge[tmp[j]]=true;
}
}
while(!que.empty()){
judge[que.front()]=false;
que.pop();
}
}
void divide(int u){
vis[u]=judge[0]=true;
solve(u);
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(vis[v]) continue;
maxp[rt=0]=sum=siz[v];
getrt(v,0);
getrt(rt,0);
divide(rt);
}
}
int main(){
scanf("%d%d",&n,&m);mem(head,-1);
for(int i=1;i<n;i++){
int u,v,w;scanf("%d%d%d",&u,&v,&w);
add(u,v,w);add(v,u,w);
}
for(int i=0;i<m;i++) scanf("%d",&q[i]);
maxp[0]=sum=n;
getrt(1,0);
getrt(rt,0);
divide(rt);
for(int i=0;i<m;i++){
if(ans[i]) puts("AYE");
else puts("NAY");
}
}