BZOJ 1589: [Usaco2008 Dec]Trick or Treat on the Farm 采集糖果

Description

每年万圣节，威斯康星的奶牛们都要打扮一番，出门在农场的N(1≤N≤100000)个牛棚里转悠，来采集糖果．她们每走到一个未曾经过的牛棚，就会采集这个棚里的1颗糖果． 农场不大，所以约翰要想尽法子让奶牛们得到快乐．他给每一个牛棚设置了一个“后继牛棚”．牛棚i的后继牛棚是Xi．他告诉奶牛们，她们到了一个牛棚之后，只要再往后继牛棚走去，就可以搜集到很多糖果．事实上这是一种有点欺骗意味的手段，来节约他的糖果．  第i只奶牛从牛棚i开始她的旅程．请你计算，每一只奶牛可以采集到多少糖果．

Input

第1行输入N，之后一行一个整数表示牛棚i的后继牛棚Xi，共N行．

Output

共N行，一行一个整数表示一只奶牛可以采集的糖果数量．

题解：

是一个基环树森林，可以用基环树的方法瞎搞。

也可以求强联通分量乱搞。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
//by zrt
//problem:
using namespace std;
],X[],P[];
int tot;
inline void add(int x,int y){
    P[++tot]=y;X[tot]=H[x];H[x]=tot;
}
];
];
];
];
];
];
];
int cnt,tim;
int top;
void tarjan(int x){
    int dfn=low[x]=++tim;
    stk[top++]=x;instk[x]=;
    if(!low[to[x]]){
        tarjan(to[x]);
        low[x]=min(low[x],low[to[x]]);
    }else if(instk[to[x]]) low[x]=min(low[x],low[to[x]]);
    if(dfn==low[x]){
        int k;
        cnt++;
        do{
            k=stk[--top];
            instk[k]=;
            belong[k]=cnt;
            hav[cnt]++;
        }while(x!=k);
    }
}
int ask(int x){
    if(ans[x]) return ans[x];
    ans[x]=hav[x];
    )
        ans[x]+=ask(P[H[x]]);
    return ans[x];
}
int main(){
    #ifdef LOCAL
    freopen("in.txt","r",stdin);
    freopen("out.txt","w",stdout);
    #endif
    int n;
    scanf("%d",&n);
    ;i<=n;i++){
        scanf("%d",&to[i]);
    }
    ;i<=n;i++){
        if(!low[i]) tarjan(i);
    }
    ;i<=n;i++){
        if(belong[i]!=belong[to[i]]) add(belong[i],belong[to[i]]);
    }
    ;i<=n;i++){
        printf("%d\n",ask(belong[i]));
    }
    ;
}