LG2921 [USACO2008DEC]Trick or Treat on the Farm 内向基环树

问题描述

LG2921

---

题解

发现一共有 \(n\) 个点，每个点只有一条出边，即只有 \(n\) 条边，于是就是一个内向基环树。

\(\mathrm{Tarjan}\) 缩点。

但是这个题比较猥琐的就是有自环。

所以断定一个强联通分量 \(i\) 是环的条件是 \(size_i>1\) 。

然后记搜求答案，特判自环。

---

\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

template <typename Tp>
void read(Tp &x){
	x=0;char ch=1;int fh;
	while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
	if(ch=='-') ch=getchar(),fh=-1;
	else fh=1;
	while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
	x*=fh;
}

const int maxn=100007;

int n;
int son[maxn],fa[maxn];

int dfn[maxn],low[maxn],ind;
bool ins[maxn];
int sta[maxn],top;

int bel[maxn],cnt;
int size[maxn];
int spe;

void tarjan(int x){
	dfn[x]=low[x]=++ind,ins[x]=1,sta[++top]=x;
	if(dfn[son[x]]){
		if(ins[son[x]]) low[x]=min(low[x],dfn[son[x]]);
	}
	else{
		tarjan(son[x]);
		low[x]=min(low[x],low[son[x]]);
	}
	if(dfn[x]==low[x]){
		++cnt;
		while(sta[top]!=x){
			bel[sta[top]]=cnt;
			ins[sta[top]]=0;--top;
			++size[cnt];
		}
		++size[cnt];
		ins[x]=0,bel[x]=cnt;--top;
	}
}

int ans[maxn];

int dfs(int x){
	if(size[bel[x]]>1) return ans[x]=size[bel[x]];
	if(ans[x]) return ans[x];
	return ans[x]=dfs(son[x])+1;
}

int main(){
	read(n);
	for(int i=1;i<=n;i++){
		read(son[i]);fa[son[i]]=i;
		if(son[i]==i) ans[i]=1;
	}
	for(int i=1;i<=n;i++){
		if(!dfn[i]) tarjan(i);
		if(size[bel[i]]>1) spe=bel[i];
	}
	for(int i=1;i<=n;i++){
		if(!ans[i]) dfs(i);
	}
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]);
	return 0;
}