洛谷 3806【模板】点分治1——点分治

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

点分治的第二道题！

和点分治的第一道题一模一样！就是多了一些询问罢了。（然而自己却写得漏洞百出，WA了数次……）

注意这回有边权为0的边，所以在 f [ sta[ i ] ] = 0 的时候可能把 f [ 0 ] 也赋成了0，要每次赋一下1。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=1e4+5,M=1e7+5,B=105,INF=0x3f3f3f3f;
int n,m,K[B],hd[N],xnt,to[N<<1],nxt[N<<1],w[N<<1];
int mn,rt,f[M],siz[N],sta[N],top;
bool vis[N],ans[B];
void add(int x,int y,int z)
{
    to[++xnt]=y;nxt[xnt]=hd[x];w[xnt]=z;hd[x]=xnt;
    to[++xnt]=x;nxt[xnt]=hd[y];w[xnt]=z;hd[y]=xnt;
}
void getrt(int cr,int fa,int s)
{
    siz[cr]=1;int mx=0;
    for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]]&&v!=fa)
    {
        getrt(v,cr,s);siz[cr]+=siz[v];mx=max(mx,siz[v]);
    }
    mx=max(mx,s-siz[cr]);
    if(mx<mn)mn=mx,rt=cr;
}
void dfs(int cr,int fa,int dis)
{
    for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]]&&v!=fa)
    {
        int s=dis+w[i];sta[++top]=s;
        for(int j=1;j<=m;j++) if(!ans[j]&&s<=K[j]&&f[K[j]-s])ans[j]=1;
        dfs(v,cr,s);
    }
}
void solve(int cr,int s)
{
    vis[cr]=1;int p0=1;f[0]=1;//说明可能有权值为0的边 
    for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]])
    {
        sta[++top]=w[i];
        for(int j=1;j<=m;j++) if(!ans[j]&&w[i]<=K[j]&&f[K[j]-w[i]])ans[j]=1;
        dfs(v,cr,w[i]);
        for(int i=p0;i<=top;i++) f[sta[i]]=1;
        p0=top+1;
    }
    for(int i=1;i<=top;i++)f[sta[i]]=0;top=0;//f[sta[i]] not f[i]!!!
    for(int i=hd[cr],v;i;i=nxt[i]) if(!vis[v=to[i]])
    {
        int ts=(siz[cr]>siz[v]?siz[v]:s-siz[cr]);//getrt时弄的siz[] 
        mn=N;getrt(v,0,ts);solve(rt,ts);
    }
}
int main()
{
    scanf("%d%d",&n,&m);int x,y,z;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%d",&x,&y,&z);add(x,y,z);
    }
    for(int i=1;i<=m;i++)scanf("%d",&K[i]);
    f[0]=1;
    mn=INF;getrt(1,0,n);solve(rt,n);
    for(int i=1;i<=m;i++) puts(ans[i]?"AYE":"NAY");
    return 0;
}