[洛谷P3806] [模板] 点分治1

洛谷 P3806 传送门

这个点分治都不用减掉子树里的了，直接搞就行了。

注意第63行 if(qu[k]>=buf[j]) 不能不写，也不能写成>。

因为这个WA了半天......

如果memset清空ex数组显然是会T的，所以开一个bef用来记录需要清空哪个地方。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 using namespace std;

 int n,m;
 int hd[],to[],nx[],len[],ec;
 int qu[],ans[];

 void edge(int af,int at,int el)
 {
     to[++ec]=at;
     len[ec]=el;
     nx[ec]=hd[af];
     hd[af]=ec;
 }

 int rt,tot;
 int sz[],mx[];
 int del[];

 void weigh(int p,int fa)
 {
     sz[p]=;mx[p]=;
     for(int i=hd[p];i;i=nx[i])
     {
         int t=to[i];
         if(t==fa||del[t])continue;
         weigh(t,p);
         sz[p]+=sz[t];
         mx[p]=max(mx[p],sz[t]);
     }
     mx[p]=max(mx[p],tot-sz[p]);
     if(mx[p]<mx[rt])rt=p;
 }

 int dis[];
 int buf[],tp,ex[],bef[],btp;

 void dfs(int p,int fa)
 {
     buf[++tp]=dis[p];
     for(int i=hd[p];i;i=nx[i])
     {
         int t=to[i];
         if(fa==t||del[t])continue;
         dis[t]=dis[p]+len[i];
         dfs(t,p);
     }
 }

 void count(int p)
 {
     btp=;
     for(int i=hd[p];i;i=nx[i])
     {
         int t=to[i];
         if(del[t])continue;
         tp=;dis[t]=len[i];
         dfs(t,p);
         for(int j=;j<=tp;j++)
             for(int k=;k<=m;k++)
                 if(qu[k]>=buf[j])
                     ans[k]|=ex[qu[k]-buf[j]];
         for(int j=;j<=tp;j++)
             bef[++btp]=buf[j],ex[buf[j]]=;
     }
     for(int i=;i<=btp;i++)ex[bef[i]]=;
 }

 void conquer(int p)
 {
     del[p]=ex[]=;
     count(p);
     for(int i=hd[p];i;i=nx[i])
     {
         int t=to[i];
         if(del[t])continue;
         rt=;tot=sz[t];
         weigh(t,);
         conquer(rt);
     }
 }

 int main()
 {
     scanf("%d%d",&n,&m);
     for(int i=;i<n;i++)
     {
         int ff,tt,vv;
         scanf("%d%d%d",&ff,&tt,&vv);
         edge(ff,tt,vv);
         edge(tt,ff,vv);
     }
     for(int i=;i<=m;i++)scanf("%d",&qu[i]);
     mx[]=tot=n;
     weigh(,);
     conquer(rt);
     for(int i=;i<=m;printf("\n"),i++)
         printf(ans[i]?"AYE":"NAY");
     return ;
 }