【NOIP 2015】消息传递（简单带权并查集）

传送门

一开始没在意数据范围...就用一个5行的dfs瞎做，搞了80分。

正解是带权并查集，只需要维护一下到根节点的距离就可以了。。。。

#include<bits/stdc++.h>
#define N 200005
using namespace std;
int n,father[N],to_root[N],ans=0x7ffffff;
inline int getfather(int x)
{
	if(father[x]==x)	return x;
	int fa=father[x];
	father[x]=getfather(father[x]);
	to_root[x]+=to_root[fa];
	return father[x];
}
inline void merge(int x,int y)
{
	int ax=getfather(x);
	int bx=getfather(y);
	if(ax!=bx){
		father[ax]=bx;
		to_root[x]=to_root[y]+1;
	}
	else ans=min(ans,to_root[x]+to_root[y]+1);
}
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)	father[i]=i;
	for(int i=1;i<=n;i++)
	{
		int x;
		cin>>x;
		merge(i,x);
	}
	cout<<ans<<endl;
	return 0;
}