【洛谷P2921】[USACO08DEC]在农场万圣节

在农场万圣节Trick or Treat on the Farm

题目链接

题解：
首先，将原图缩点，变为DAG，

然后在DAG上记忆化搜索即可

 #include<iostream>
 #include<cstring>
 #include<cstdio>
 using namespace std;
 #define N 100010
 #define reset(a) memset(a,0,sizeof(a))
 int n,next[N],belong[N],num;
 int Next[N],f[N],size[N];
 int dfn[N],low[N],cnt;
 int stack[N],top;
 bool ins[N];
 void dfs(int t){
     dfn[t]=low[t]=++cnt;
     stack[++top]=t;
     ins[t]=;
     int v=next[t];
     if(!dfn[v]){
         dfs(v);
         low[t]=min(low[t],low[v]);
     }
     else if(ins[v])
         low[t]=min(low[t],dfn[v]);
     if(low[t]==dfn[t]){
         num++;
         size[num]++;
         while(stack[top]!=t){
             size[num]++;
             int u=stack[top];
             belong[u]=num;
             ins[u]=;
             top--;
         }
         top--;
         ins[t]=;
         belong[t]=num;
     }
 }
 int Dfs(int t){
     if(f[t]) return f[t];
     if(!Next[t]) return f[t]=size[t];
     return f[t]=size[t]+Dfs(Next[t]);
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;i++)
      scanf("%d",&next[i]);
     for(int i=;i<=n;i++)
      if(!dfn[i]) dfs(i);
     for(int i=;i<=n;i++)
      if(belong[i]!=belong[next[i]])
       Next[belong[i]]=belong[next[i]];
     for(int i=;i<=n;i++){
         if(!f[belong[i]])
          Dfs(belong[i]);
         printf("%d\n",f[belong[i]]);
     }
     return ;
 }