Tarjan模板【洛谷P2921】

传送门：https://www.luogu.org/problemnew/show/P2921

上代码先占坑，我去打打NOIP2018(滑稽)

好了好了回来了，NOIP太难了，做不动。

Tarjan是用来求强连通分量的算法。应用十分广泛。

Tarjan算法就不证明了，有兴趣的可以去网上看大神们的各种证明。(主要是我也不会证明啊)

随便讲讲Tarjan怎么写的吧。

low[]和dfn[]是最重要的两个数组。

low[]作为每个点在这颗树中的，最小的子树的根，每次保证最小。dfn[]是时间戳，记录now结点从起点访问的步数。

还要有一个stack<int> s记录结点，如果栈里面有两个相同的结点，表示他们之间的结点全部都是强连通分量的结点。

%染色%

既然Tarjan可以求强连通分量，那么我们就可以把图中的环全部染成相同的颜色，颜色相同的一堆看作一个结点。这个过程叫做缩点。然后就可以把有环的图转化成无环的图啦。

那么如果要求图中的最小环，那不就是图中的染色面积最小的那个嘛(不算1)。

---

讲讲这个题吧，快忘记这个题目的存在了。

对于每一头牛，如果这头牛在环里，那么他的结果就是环的大小。

如果这头牛不在环里，那么就是这头牛到环的距离+环的大下。

main函数里写了这么多for，相信大家可以看懂(也没人看我的题解吧)。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+7;
int nxt[maxn];
int ans[maxn];
int color[maxn];
int sum[maxn],vis[maxn];
int low[maxn],dfn[maxn];
vector<int>G[maxn];
stack<int> s;
int cnt = 0;
int col = 0;
void tarjan(int x)
{
	low[x] = dfn[x] = ++cnt;
	s.push(x);
	vis[x] = 1;
	for(int i=0;i<G[x].size();i++)
	{
		int v = G[x][i];
		if(dfn[v]==0)
		{
			tarjan(v);
			low[x] = min(low[x],low[v]);
		}
		else if(vis[v]==1)
		{
			low[x] = min(low[x],dfn[v]);
		}
	}
	if(low[x]==dfn[x])
	{
		col++;
		while(1)
		{
			int tmp = s.top();
			s.pop();
			color[tmp] = col;
			vis[tmp] = -1;
			if(tmp==x) break;
		}
	}
}
void search(int s,int e,int stp)
{
	if(ans[e])
	{
		ans[s] = ans[e]+stp;
		return;
	}
	search(s,nxt[e],stp+1);
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>nxt[i];
		G[i].push_back(nxt[i]);
		if(nxt[i]==i) ans[i] = 1;
	}
	for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			tarjan(i);
		}
	}
	for(int i=1;i<=n;i++)
	{
		sum[color[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) search(i,nxt[i],1);
	}
	for(int i=1;i<=n;i++)
	{
		cout<<ans[i]<<endl;
	}
	return 0;
}