【BZOJ1589】【USACO 2008 Dec Gold】 1.Trick or Treat on the Farm 基环树裸DP

【BZOJ1589】【USACO 2008 Dec Gold】 1.Trick or Treat on the Farm 基环树裸DP、

没测样例一遍A这真是……

题意：每个点都有且仅有一个出边（可以出现自环），然后这样一个点出发就会走过且一定走过某些点。

问每个点出发都会走过几个点。

首先这是基环树无疑。

然后就是裸DP了。

这个的关键就是找环，仅此。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 101000
using namespace std;
int next[N],f[N];
int stk[N],top,n;
int vis[N];
int main()
{
	register int i,j,k;
	scanf("%d",&n);
	for(i=1;i<=n;i++)scanf("%d",&next[i]);
	for(i=1;i<=n;i++)if(!vis[i])
	{
		top=0;
		for(j=i;!vis[j];j=next[j])
		{
			stk[++top]=j;
			vis[j]=i;
		}
		if(vis[j]==i)
		{
			int tt=top;
			while(stk[tt--]!=j);
			tt=top-tt;
			do{
				f[stk[top--]]=tt;
			}while(stk[top+1]!=j);
		}
		k=0;
		while(top)f[stk[top--]]=f[j]+(++k);
	}
	for(i=1;i<=n;i++)printf("%d\n",f[i]);
	return 0;
}