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 3010
int 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;
}