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;

}