六亲不认的代码

#include<bits/stdc++.h>
using namespace std;
int n,next[100001],f[100001];
int vis[100001],T=0,P,h[100001];
void dfs(int k,int deep){
	if(h[k]) return;
	if(vis[k]){
		T=k;P=deep-vis[k];
		f[k]=P;h[k]=1;
		return;
	}
	vis[k]=deep;
	dfs(next[k],deep+1);
	h[k]=1;
	vis[k]=0;
	if(T&&k==T){
		f[k]=P;T=0;P=0;h[k]=1;
		return;
	}
	if(T){
		h[k]=1;f[k]=P;return;
	}
	return;
}
int dp(int k){
	if(f[k]) return f[k];
	return f[k]=dp(next[k])+1;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i) scanf("%d",&next[i]),f[i]=0;
	memset(vis,0,sizeof(vis));
	memset(h,0,sizeof(h));
	for(register int i=1;i<=n;++i)
		if(h[i]) continue;
		else dfs(i,1);
	for(int i=1;i<=n;++i)
	    printf("%d\n",dp(i));
	return 0;
}