模板点分治

Posted 2020-11-01 Driver_Lao

洛谷 3806 

 1 #include<cstdio>
 2 #include<algorithm>
 3 #define N 10010
 4 #define rg register
 5 using namespace std;
 6 int n,m,tot,cnt,root,last[N],siz[N],mxsiz[N],dep[N],q[N];
 7 bool have[10000010],ans[110],v[N];
 8 struct edge{
 9     int to,pre,dis;
10 }e[N<<2];
11 inline int read(){
12     int k=0,f=1; char c=getchar();
13     while(c<\'0\'||c>\'9\')c==\'-\'&&(f=-1),c=getchar();
14     while(\'0\'<=c&&c<=\'9\')k=k*10+c-\'0\',c=getchar();
15     return k*f;
16 }
17 void getroot(int x,int fa){
18     siz[x]=1; 
19     for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]&&to!=fa){
20         getroot(to,x); 
21         siz[x]+=siz[to]; mxsiz[x]=max(mxsiz[x],siz[to]);
22     }
23     mxsiz[x]=max(mxsiz[x],cnt-siz[x]);
24     if(!root||mxsiz[x]<mxsiz[root]) root=x;
25 }
26 void getdep(int x,int fa,int d){
27     dep[tot++]=d;
28     for(rg int i=last[x],to;i;i=e[i].pre)
29     if(!v[to=e[i].to]&&to!=fa) getdep(to,x,d+e[i].dis);
30 }
31 void dfs(int x){
32     v[x]=1; int l=tot=1; have[0]=1;
33     for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]){
34         l=tot; getdep(to,x,e[i].dis);
35         for(rg int j=1;j<=m;j++){
36             if(ans[j]) continue;
37             for(rg int k=l;k<tot;k++)if(q[j]>=dep[k]&&have[q[j]-dep[k]]){
38                 ans[j]=1; break;
39             }
40         }
41         for(rg int j=l;j<tot;j++) have[dep[j]]=1;
42     }
43     for(rg int i=1;i<tot;i++) have[dep[i]]=0;
44     for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]){
45         cnt=siz[to]; root=0; getroot(to,x); dfs(root);
46     }
47 }
48 int main(){
49     n=read(); m=read();
50     for(rg int i=1;i<n;i++){
51         int u=read(),v=read(),w=read();
52         e[++tot]=(edge){v,last[u],w}; last[u]=tot;
53         e[++tot]=(edge){u,last[v],w}; last[v]=tot;
54     }
55     for(rg int i=1;i<=m;i++) q[i]=read();
56     root=0; cnt=n; getroot(1,0); dfs(root);
57     for(rg int i=1;i<=m;i++) puts(ans[i]?"AYE":"NAY");
58     return 0;
59 }

View Code