BZOJ1316 树上的询问
题目传送门
分析:
点分治板题,对于每个重心,set维护每个子树里面的路径,然后每添加一棵子树的值之前,询问一下set里面是否有Len-dis[i]即可
找到了就不做了,可以变很快
注意询问的Len可能为0
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#define maxn 20005
using namespace std;
inline int getint()
{
int num=0,flag=1;char c;
while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
while(c>='0'&&c<='9')num=num*10+c-48,c=getchar();
return num*flag;
}
int n,q,K,rt,N;
int fir[maxn],nxt[maxn],to[maxn],len[maxn],cnt;
int qq[maxn];
int sz[maxn];
bool vis[maxn];
int a[maxn],cur,dis[maxn];
set<int>S;
set<int>::iterator it;
bool ans[maxn];
inline void newnode(int u,int v,int w)
{to[++cnt]=v,nxt[cnt]=fir[u],fir[u]=cnt,len[cnt]=w;}
inline void getrt(int u,int fa)
{
sz[u]=1;
int mx=0;
for(int i=fir[u];i;i=nxt[i])
if(to[i]!=fa&&!vis[to[i]])
{
getrt(to[i],u);
sz[u]+=sz[to[i]];
mx=max(mx,sz[to[i]]);
}
mx=max(mx,N-sz[u]);
if(mx<=N/2)rt=u;
}
inline void dfs(int u,int fa)
{
sz[u]=1;
for(int i=fir[u];i;i=nxt[i])
if(to[i]!=fa&&!vis[to[i]])
dfs(to[i],u),sz[u]+=sz[to[i]];
}
inline void cal(int rt)
{
S.clear();S.insert(0);
for(int i=fir[rt];i;i=nxt[i])
if(!vis[to[i]])
{
cur=0;
queue<int>Q;
dis[to[i]]=len[i];
Q.push(to[i]);a[++cur]=dis[to[i]];
while(!Q.empty())
{
int u=Q.front();Q.pop();
for(int j=fir[u];j;j=nxt[j])
if(sz[u]>sz[to[j]])
dis[to[j]]=dis[u]+len[j],Q.push(to[j]),a[++cur]=dis[to[j]];
}
for(int k=1;k<=q;k++)if(!ans[k])for(int j=1;j<=cur;j++)if(S.find(qq[k]-a[j])!=S.end())ans[k]=1;
for(int j=1;j<=cur;j++)S.insert(a[j]);
}
}
inline void solve(int u)
{
for(int i=fir[rt];i;i=nxt[i])
if(!vis[to[i]])
{
N=sz[to[i]];
getrt(to[i],to[i]);
dfs(rt,rt),cal(rt),vis[rt]=1;
solve(to[i]);
}
}
int main()
{
n=getint(),q=getint();
for(int i=1;i<n;i++)
{
int u=getint(),v=getint(),w=getint();
newnode(u,v,w),newnode(v,u,w);
}
for(int i=1;i<=q;i++)qq[i]=getint();
N=n;getrt(1,1);
dfs(rt,rt),cal(rt);vis[rt]=1;
solve(rt);
for(int i=1;i<=q;i++)
if(!qq[i]||ans[i])printf("Yes\n");
else printf("No\n");
}
版权声明:本文为Darknesses原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。