正題

評測記錄:https://www.luogu.org/recordnew/lists?uid=52918&pid=P3806

---

題目大意

一顆樹，k個詢問，求是否存在長度為$l_i$的路徑。

---

解題思路

開個桶就好了，點分治不解釋。

---

$code$

#include<cstdio> #include<algorithm> #include<cstring> #define N 100010 #define inf 10000000 using namespace std; struct node{int to,next,w; }a[N*2]; int n,m,que[1010],ok[inf],tot,sum,num,q[N]; int root,siz[N],f[N],ls[N],st[N],dis[N]; int lon[inf]; bool v[N]; void addl(int x,int y,int w) {a[++tot].to=y;a[tot].next=ls[x];a[tot].w=w;ls[x]=tot; } void groot(int x,int fa) {siz[x]=1;f[x]=0;for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(y==fa||v[y]) continue;groot(y,x);siz[x]+=siz[y];f[x]=max(f[x],siz[y]);}f[x]=max(f[x],sum-siz[x]);if(f[x]<f[root]) root=x; } void get_dis(int x,int fa) {st[++num]=dis[x];for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(y==fa||v[y]) continue;dis[y]=dis[x]+a[i].w;get_dis(y,x);} } void calc(int x) {int p=0;for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(v[y]) continue;num=0;dis[y]=a[i].w;get_dis(y,x);for(int j=num;j;j--)for(int k=1;k<=m;k++)ok[k]|=lon[que[k]-st[j]];for(int j=num;j;j--)q[++p]=st[j],lon[st[j]]=1;}for(int i=1;i<=p;i++)lon[q[i]]=0; } void solve(int x) {v[x]=lon[0]=1;calc(x);for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(v[y]) continue;sum=siz[y];f[root=0]=n;groot(y,0);solve(y);} } int main() {scanf("%d%d",&n,&m);for(int i=1;i<n;i++){int x,y,w;scanf("%d%d%d",&x,&y,&w);addl(x,y,w);addl(y,x,w);}for(int i=1;i<=m;i++)scanf("%d",&que[i]);f[0]=n;sum=n;groot(1,0);solve(root);for(int i=1;i<=m;i++)if(ok[i]) printf("AYE\n");else printf("NAY\n"); }

總結

以上是生活随笔為你收集整理的P3806-【模板】点分治1的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。