洛谷P3806（点分治）

题目链接：https://www.luogu.org/problemnew/show/P3806

思路：点分治模板题

#pragma GCC optimize(2)
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <bitset>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include <iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod ll(1000)
#define pb push_back
#define eps 1e-8
#define lc d<<1
#define rc d<<1|1
#define Pll pair<ll,ll>
#define P pair<int,int>
#define pi acos(-1)
int f[50008],siz[50008],tot,head[50008],rt,n,m,k,vis[50008],dis[50008],qw,a[50008],query[108],ans[108],b[50008];
bool ds[10000008];
struct node{
int next,to,v;}e[50008];
void add(int a,int b,int c)
{
    e[tot].next=head[a];
    e[tot].to=b;
    e[tot].v=c;
    head[a]=tot++;
}
void getro(int x,int fa)
{
    f[x]=0,siz[x]=1;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int y=e[i].to;
        if(y==fa||vis[y]) continue;
        getro(y,x);
        f[x]=max(f[x],siz[y]);
        siz[x]+=siz[y];
    }
    f[x]=max(f[x],n-siz[x]);
    if(f[x]<f[rt]) rt=x;
}
void get_dis(int x,int fa)
{
    dis[++qw]=a[x];
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int y=e[i].to;
        if(y==fa||vis[y]) continue;
        a[y]=e[i].v+a[x];
        get_dis(y,x);
    }
}
void slove(int x,int fa)
{
    int num=0;ds[0]=1;
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int y=e[i].to;
        if(y==fa||vis[y]) continue;
        qw=0,a[y]=e[i].v;
        get_dis(y,x);
        FOR(j,1,qw)
         FOR(k,1,m)
          if(query[k]>=dis[j])  ans[k]|=ds[query[k]-dis[j]];
        FOR(j,1,qw) ds[dis[j]]=1,b[++num]=dis[j];
    }
    FOR(i,1,num) ds[b[i]]=0;
}
void Divide(int x)
{
    vis[x]=1;
    slove(x,0);
    for(int i=head[x];i!=-1;i=e[i].next)
    {
        int y=e[i].to;
        if(vis[y]) continue;
        n=siz[x],rt=0,f[0]=inf;
        getro(y,x);
        Divide(rt);
    }
}
int main()
{
    cin.tie(0);
    cout.tie(0);
    REW(head,-1);
    cin>>n>>m;
    int x,y,z;
    FOR(i,1,n-1)
    {
        si(x),si(y),si(z);
        add(x,y,z);
        add(y,x,z);
    }
    FOR(i,1,m) si(query[i]);
    f[0]=inf;
    getro(1,0);
    Divide(rt);
    FOR(i,1,m)
    {
        if(ans[i]) puts("AYE");
        else puts("NAY");
    }
    return 0;
}