(Luogu) P2921 [USACO08DEC]Trick or Treat on the Farm( Tarjan求强联通分量 )

传送门

虽然看题解很多大佬并没有用Tarjan来做这一题，但是这个问题用Tarjan求强联通分量来解决这题还是很清晰的，如果一头牛在一个强联通分量里，那它可以拿的糖果就是这个强联通分量的点数。反之由于牛走的的方向只有一个，那就直接深搜到第一个强联通分量，加上这个强联通分量的点数就是它可以拿到的糖果。

跑强联通分量的同时，就可以知道某个点属于哪一块，并且一块有多大。

注：自环的点要先处理一下，ans为1，和可以深搜到大强联通分量的分开来，方便写深搜。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+7;
int low[maxn],dfn[maxn],cnt=0,tot=0;
int color[maxn],sum[maxn],n;
int ans[maxn];
bool added[maxn];
vector<int> G[maxn];
stack<int> s;

void tarjan(int x){
	low[x]=dfn[x]=++cnt;
	s.push(x);
	added[x]=1;
	for(int i=0;i<(int)G[x].size();++i){
		int np=G[x][i];
		if(!dfn[np]){
			tarjan(np);
			low[x]=min(low[x],low[np]);
		}
		else if(added[np]){
			low[x]=min(low[x],dfn[np]);
		}
	}

	if(low[x]==dfn[x]){
		tot++;
		while(1){
			int tmp=s.top();s.pop();
			color[tmp]=tot,sum[tot]++;
			added[tmp]=false;
			if(tmp==x)	break;
		}
	}
}

void dfs(int point,int np,int step){
	if(ans[np]){
		ans[point]=ans[np]+step;
		return ;
	}
	for(int i=0;i<(int)G[np].size();++i){
		dfs(point,G[np][i],step+1);
	}

}

int main(){
	memset(added,false,sizeof(added));
	cin>>n;
	int t;
	for(int i=1;i<=n;++i){
		cin>>t;
		G[i].push_back(t);
		if(i==t)	ans[i]=1;
	}
	for(int i=1;i<=n;++i){
		if(!dfn[i])	tarjan(i);
	}
	for(int i=1;i<=n;++i){
		if(sum[color[i]]!=1)	ans[i]=sum[color[i]];
	}
	for(int i=1; i<=n; i++) {
		if(ans[i]==0) dfs(i,i,0);
	}
	for(int i=1;i<=n;++i){
		cout<<ans[i]<<endl;
	}
	return 0;
}