点分治

1

点分治好难写呀
写的变量好多，太乱了，一点也不优美

代码

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+7;
int read() {
    int x=0,f=1;char s=getchar();
    for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
    for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
    return x*f;
}
int n,m,rt,rt_rt,tot;
struct node {
    int u,v,q,nxt;
}e[N<<1];
int head[N<<1],cnt;
void add(int u,int v,int q) {
    e[++cnt].v=v;
    e[cnt].q=q;
    e[cnt].nxt=head[u];
    head[u]=cnt;
}
int dis[N],siz[N],Q[N],stak[N],q[N];
bool pd[10000007],vis[N],ok[N];
void getrt(int u,int f) {
    siz[u]=1;
    int tmp=0;
    for(int i=head[u];i;i=e[i].nxt) {
        int v=e[i].v;
        if(v==f||vis[v]) continue;
        getrt(v,u);
        siz[u]+=siz[v];
        tmp=max(tmp,siz[v]);
    }
    tmp=max(tmp,tot-siz[u]);
    if(tmp<rt_rt) rt_rt=tmp,rt=u;
}
void getdis(int u,int f) {
    stak[++stak[0]]=dis[u];
    for(int i=head[u];i;i=e[i].nxt) {
        int v=e[i].v;
        if(v==f||vis[v]) continue;
        dis[v]=dis[u]+e[i].q;
        getdis(v,u);
    }
}

void calc(int u) {
    int p=0;
    for(int i=head[u];i;i=e[i].nxt) {
        int v=e[i].v;
        if(vis[v]) continue;
        stak[0]=0,dis[v]=e[i].q;
        getdis(v,u);
        for(int j=1;j<=stak[0];++j) {
            for(int k=1;k<=m;++k) {
                if(Q[k]>=stak[j]&&pd[Q[k]-stak[j]]) ok[k]=1;
            }
        }
        for(int j=1;j<=stak[0];++j) {
            q[++p]=stak[j],pd[stak[j]]=1;
        }
    }
    for(int i=1;i<=p;++i) pd[q[i]]=0;
}
void dfs(int u) {
    pd[0]=vis[u]=1,calc(u);
    for(int i=head[u];i;i=e[i].nxt) {
        int v=e[i].v;
        if(vis[v]) continue;
        tot=siz[v],rt_rt=0x3f3f3f3f,rt=0;
        getrt(v,0);
        dfs(rt);
    }
}
int main() {
    n=read(),m=read();
    for(int i=1;i<n;++i) {
        int x=read(),y=read(),z=read();
        add(x,y,z);
        add(y,x,z);
    }
    for(int i=1;i<=m;++i) Q[i]=read();
    rt_rt=tot=n;
    getrt(1,0);
    dfs(rt);
    for(int i=1;i<=m;++i) {
        if(ok[i]) puts("AYE");
        else puts("NAY");
    }
    return 0;
}