首页 > 代码库 > BZOJ4122 : [Baltic2015]File paths

BZOJ4122 : [Baltic2015]File paths

对于在$o$点的某个询问,有两种情况:

 

情况1:走到任意一个点$x$然后超链接跳到$o$的某个祖先$y$再走到$o$。

枚举所有$y$看看是否存在$x$即可。

时间复杂度$O(nm)$。

 

情况2:走到$o$的某个祖先$x$,然后走到$x$子树内某个点$y$,在$x$和$y$之间不断通过超链接来回走,最后再从$x$走到$o$。

枚举$x$和$y$维护出每种环长是否可能存在,然后枚举约数判断是否存在对应的$x$和$y$即可。

时间复杂度$O(n^2+md(k))$。

 

#include<cstdio>#define N 3010int n,m,K,S,i,x,a[N],len[N],g[N],G[N],v[N<<1],nxt[N<<1],ed;int q[N],t,st[N],en[N],dfn,seq[N],f[2000010];bool ans[N];inline void add(int&x,int y){v[++ed]=y;nxt[ed]=x;x=ed;}void dfs1(int x){  q[++t]=a[x];  seq[st[x]=++dfn]=a[x];  for(int i=G[x];i;i=nxt[i]){    int o=v[i];    for(int j=1;j<=t;j++){      int k=K-S-a[x]-len[o]+q[j];      if(k>=0&&k<=K)if(f[k]){ans[o]=1;break;}    }  }  for(int i=g[x];i;i=nxt[i])dfs1(v[i]);  t--;  en[x]=dfn;}void dfs2(int x){  for(int i=st[x];i<=en[x];i++)f[seq[i]-a[x]+S]++;  for(int i=G[x];i;i=nxt[i]){    int o=v[i];    if(ans[o])continue;    int k=K-a[x]-len[o];    for(int j=1;j*j<=k;j++)if(k%j==0)if(f[j]||f[k/j]){ans[o]=1;break;}  }  for(int i=g[x];i;i=nxt[i])dfs2(v[i]);  for(int i=st[x];i<=en[x];i++)f[seq[i]-a[x]+S]--;}int main(){  scanf("%d%d%d%d",&n,&m,&K,&S);S++;  for(i=1;i<=n;i++){    scanf("%d%d",&x,&a[i]);    a[i]+=a[x]+1;    add(g[x],i);  }  for(i=1;i<=m;i++){    scanf("%d%d",&x,&len[i]);    len[i]++;    ans[i]=a[x]+len[i]==K;    if(!ans[i])add(G[x],i);  }  for(i=0;i<=n;i++)f[a[i]]=1;  dfs1(0);  for(i=0;i<=n;i++)f[a[i]]=0;  dfs2(0);  for(i=1;i<=m;i++)puts(ans[i]?"YES":"NO");  return 0;}

  

BZOJ4122 : [Baltic2015]File paths