[POI2005]SKA-Piggy Banks tarjan 水题

Code:

#include<bits/stdc++.h>
#define maxn 1000002 
using namespace std;
void setIO(string s)
{
	string in=s+".in"; 
	freopen(in.c_str(),"r",stdin); 
}                           
stack<int>S; 
int scc,cnt,edges; 
int low[maxn],pre[maxn],vis[maxn],bin[maxn],du[maxn];  
vector<int>G[maxn];         
void tarjan(int u)
{
	S.push(u); 
	low[u]=pre[u]=++scc; 
	vis[u]=1; 
	for(int i=0,sz=G[u].size();i<sz;++i)
	{
		int v=G[u][i];          
		if(!vis[v]) tarjan(v), low[u]=min(low[u],low[v]); 
		else if(vis[v]==1) low[u]=min(low[u], pre[v]); 
	}
	if(low[u]==pre[u])
	{
		++cnt; 
		for(;;)
		{
			int x=S.top(); S.pop(); 
			bin[x]=cnt, vis[x]=-1; 
			if(x==u) break; 
		}
	}
}
int main()
{
	// setIO("input"); 
	int n; 
	scanf("%d",&n); 
	for(int i=1;i<=n;++i)
	{
		int a; 
		scanf("%d",&a); 
		G[a].push_back(i); 
	}
	for(int i=1;i<=n;++i) 
		if(!vis[i]) 
			tarjan(i);
	for(int i=1;i<=n;++i) 
	{
		for(int j=0,sz=G[i].size();j<sz;++j) 
		{
			if(bin[i]!=bin[G[i][j]]) ++du[bin[G[i][j]]];                            
		}
	}
	int ans=0;   
	for(int i=1;i<=cnt;++i) if(!du[i]) ++ans; 
	printf("%d\n",ans);   
	return 0; 
}