luogu 3806
Posted Stump
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了luogu 3806相关的知识,希望对你有一定的参考价值。
https://www.luogu.org/problemnew/show/P3806
点分治棵题
对于每次询问,离线处理,每次取重心点分治,查询到每个位置,对于每次询问进行查询.
#include<stdio.h> #include<algorithm> #include<set> #define FOR(i,s,t) for(register int i=s;i<=t;++i) #define ROF(i,s,t) for(register int i=s;i>=t;--i) #define VIS(now) for(register int e=las[now];e;e=nxt[e]) using std::set; const int N=10011,MAXX=10000011; int heap[N],keep[N]; bool vis[N],ans[N]; int n,m,tot,k; int nxt[N<<1],las[N],to[N<<1],q[N],sz[N],w[N<<1],dis[N<<1]; set<int>s; int x,y,z; inline void add(int x,int y,int z){ nxt[++tot]=las[x];las[x]=tot;to[tot]=y;w[tot]=z; } inline void dfs_size(int x,int p,int &mn,int &G){ sz[x]=1; int maxx=0; VIS(x) if(to[e]!=p&&!vis[to[e]]){ dfs_size(to[e],x,mn,G); sz[x]+=sz[to[e]]; if(maxx<sz[to[e]]) maxx=sz[to[e]]; } if(maxx<n-sz[x]) maxx=n-sz[x]; if(maxx<mn) mn=maxx,G=x; } inline int getG(int x){ int mn=n,G=0; dfs_size(x,0,mn,G); return G; } inline void dfs_path(int x,int p,int len){ if(len>k)return; keep[++keep[0]]=len; FOR(i,1,m) if(s.count(q[i]-len))ans[i]=1; VIS(x) if(to[e]!=p&&!vis[to[e]]) dfs_path(to[e],x,len+w[e]); } inline void find(int x){ int G=getG(x); s.clear(); s.insert(0); vis[G]=1; VIS(G) if(!vis[to[e]]){ keep[0]=0; dfs_path(to[e],G,w[e]); FOR(i,1,keep[0]) s.insert(keep[i]); } VIS(G) if(!vis[to[e]]) find(to[e]); } int main(){ scanf("%d%d",&n,&m); FOR(i,2,n){ scanf("%d%d%d",&x,&y,&z); add(x,y,z); add(y,x,z); } FOR(i,1,m){ scanf("%d",q+i); if(q[i]>k)k=q[i]; } find(1); FOR(i,1,m) ans[i]?puts("AYE"):puts("NAY"); return 0; }
以上是关于luogu 3806的主要内容,如果未能解决你的问题,请参考以下文章