题目 https://www.luogu.org/problemnew/show/P3806
题意 模板题
思路 模板题
#include <bits/stdc++.h>
using namespace std;
const int inf = 10000000;
const int maxn = 100010;
int n,m;
struct node
{
int v,dis,nxt;
}E[maxn<<1];
int tot,head[maxn];
int maxp[maxn],size[maxn];//当前节点最大子树节点数
int dis[maxn],rem[maxn],judge[maxn]; //根节点当当前节点长度 当前根节点到某子树所以子节点长度 当前根节点之前所以子树出现的长度
int vis[maxn];//当前节点是否已被分治
int test[maxn],q[maxn];//记录答案 记录出现的长度
int query[1100];
int sum,rt,ans;
void add(int u,int v,int dis)
{
E[++tot].nxt = head[u];
E[tot].v = v;
E[tot].dis = dis;
head[u] = tot;
}
void getrt(int u,int pa) //查询重心
{
size[u] = 1;
maxp[u] = 0;
for(int i = head[u];i != -1;i = E[i].nxt)
{
int v = E[i].v;
if(v == pa||vis[v]) continue;
getrt(v,u);
size[u] += size[v];
maxp[u] = max(maxp[u],size[v]);
}
maxp[u] = max(maxp[u],sum-size[u]);
if(maxp[u] < maxp[rt]) rt = u;
}
void getdis(int u,int fa) //查询节点到当前根的长度
{
rem[++rem[0]] = dis[u];
for(int i = head[u];i != -1;i=E[i].nxt)
{
int v = E[i].v;
if(v == fa||vis[v]) continue;
dis[v] = dis[u] + E[i].dis;
getdis(v,u);
}
}
void calc(int u) //当前根节点出现的结果
{
int p = 0;
for(int i = head[u];i != -1;i=E[i].nxt)
{
int v = E[i].v;
if(vis[v]) continue;
rem[0] = 0;
dis[v] = E[i].dis;
getdis(v,u);
for(int j = rem[0];j >= 1;j--)
{
for(int k = 1;k <= m;k++)
{
if(query[k] >= rem[j])
{
test[k]|=judge[query[k]-rem[j]];
}
}
}
for(int j = rem[0];j >= 1;j--)
{
q[++p] = rem[j],judge[rem[j]] = 1;
}
}
for(int i = 1;i <= p;i++)
{
judge[q[i]] = 0;
}
}
void solve(int u)
{
vis[u] = judge[0] = 1;
calc(u);
for(int i = head[u];i != -1;i = E[i].nxt)
{
int v = E[i].v;
if(vis[v]) continue;
sum = size[v],maxp[rt = 0] = inf;
getrt(v,0),solve(rt);
}
}
int main()
{
memset(head,-1,sizeof(head));
tot = 0;
scanf("%d%d",&n,&m);
for(int i = 1;i < n;i++) // 建图
{
int u,v,dis;
scanf("%d%d%d",&u,&v,&dis);
add(u,v,dis),add(v,u,dis);
}
for(int i = 1;i <= m;i++)
{
scanf("%d",&query[i]);
}
maxp[rt] = sum = n;
getrt(1,0); //求重心
memset(vis,0,sizeof(vis));
solve(rt);//开始分治
for(int i =1;i <= m;i++)
{
if(test[i]) printf("AYE\n");
else printf("NAY\n");
}
return 0;
}