标签：++ base play 说明 hit mes names dfs ttl
感谢hzwer的点分治互测。
给定一棵有n个点的树
询问树上距离为k的点对是否存在。
n,m 接下来n-1条边a,b,c描述a到b有一条长度为c的路径
接下来m行每行询问一个K
输出格式：对于每个K每行输出一个答案，存在输出“AYE”,否则输出”NAY”(不包含引号)
2 1 1 2 2 2
AYE
对于30%的数据n<=100
对于60%的数据n<=1000,m<=50
对于100%的数据n<=10000,m<=100,c<=10000,K<=10000000
对于每个点，分为有长度为k的路径经过点以及长度为k的路径在其子树
对于长度为k的路径在其子树的点，我们可以将其看作是长度为k的路径经过其子树所在根节点
那么这就是点分治的基本思想
为降低复杂度，我们每次采取重心作为根节点。将树的深度将至logn
则复杂度为O（NlogN2)
#include<bits/stdc++.h> #define inf 0x3f3f3f3f using namespace std; const int maxn=10000+100,maxm=10000000+100; int head[maxn]; int q[maxn],ans[maxn]; int p[maxm],t[maxn]; int d[maxn],dis[maxn],vis[maxn]; int s[maxn],ms[maxn]; int rt=0,sum; int n,m; struct edge { int to,next,val; }e[maxn<<1]; int size=0; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();} while(ch>=‘0‘&&ch<=‘9‘){x=(x<<3)+(x<<1)+ch-‘0‘;ch=getchar();} return x*f; } void addedge(int u,int v,int w) { e[++size].to=v;e[size].val=w;e[size].next=head[u];head[u]=size; } void tc(int u,int fa)//树的重心 { s[u]=1;ms[u]=0; for(int i=head[u];i;i=e[i].next) { int to=e[i].to; if(to==fa||vis[to])continue; tc(to,u); s[u]+=s[to]; ms[u]=max(ms[u],s[to]); } ms[u]=max(ms[u],sum-ms[u]); if(ms[u]<ms[rt])rt=u; } void dfs(int u,int fa) { d[++d[0]]=dis[u]; for(int i=head[u];i;i=e[i].next) { int to=e[i].to; if(to==fa||vis[to])continue; dis[to]=dis[u]+e[i].val; dfs(to,u); } } void cal(int u) { int tot=0; for(int i=head[u];i;i=e[i].next) { int to=e[i].to; if(vis[to])continue; d[0]=0;dis[to]=e[i].val; dfs(to,u); for(int j=1;j<=d[0];++j) for(int k=1;k<=m;++k) if(!ans[k]&&q[k]>=d[j])//有长度为q[k]的路径:该子树与之前其他子树距离和为q[k] ans[k]=p[q[k]-d[j]]; for(int j=1;j<=d[0];++j) t[++tot]=d[j],p[d[j]]=1;//用桶存下到该节点已有的距离 } for(int i=1;i<=tot;i++) p[t[i]]=0;//桶清空，用memsetTLE } void solve(int u) { vis[u]=p[0]=1;cal(u); for(int i=head[u];i;i=e[i].next) { int to=e[i].to; if(vis[to])continue; sum=s[to],ms[rt=0]=inf; tc(to,0);solve(rt); } } int main() { n=read(),m=read(); for(int i=1;i<n;i++) { int u=read(),v=read(),w=read(); addedge(u,v,w); addedge(v,u,w); } for(int i=1;i<=m;i++) q[i]=read(); sum=n;ms[rt]=inf; tc(1,0); solve(rt); for(int i=1;i<=m;i++) puts(ans[i]?"AYE":"NAY"); return 0; }
标签：++ base play 说明 hit mes names dfs ttl
原文地址：https://www.cnblogs.com/DriverBen/p/10963438.html
