BZOJ 1529 [POI2005]ska Piggy banks 并查集

Description

Byteazar 有 N 个小猪存钱罐. 每个存钱罐只能用钥匙打开或者砸开. Byteazar 已经把每个存钱罐的钥匙放到了某些存钱罐里. Byteazar 现在想买一台汽车于是要把所有的钱都取出来. 他想尽量少的打破存钱罐取出所有的钱,问最少要打破多少个存钱罐.

Input

第一行一个整数 N (1 <= N <= 1.000.000) – 表示存钱罐的总数. 接下来每行一个整数,第 i+1行的整数代表第i个存钱罐的钥匙放置的存钱罐编号.

Output

一个整数表示最少打破多少个存钱罐.

Sample Input

4
2
1
2
4

Sample Output

2
In the foregoing example piggy banks 1 and 4 have to be smashed.

HINT

传送门

看到题之后觉得sb题= =然后写了个tarjan。。

tarjan的思路很简单，假设x[i]是i的钥匙存放的钱罐，那么i向x[i]连一条边，

然后tarjan缩点，再将新的DAG的所有边反向跑拓扑就好了。

因为对于一个强连通分量，只要一个打破其它都通了。

然后完美MLE= =

不明所以……运行内存那么大吗？= c =

正解是并查集……怎么来的呢？

观察一下前面tarjan的方法，

可以看到之前所说“i向x[i]连一条边”，那么每一个点都只有一个出边。

也就是说，一个点要么在某个强连通分量里，要么连向某个强连通分量或点。

如果这个点连向某个强连通分量或点，

刚才的方法里，缩点之后把所有边反向然后不断找入度为0，

也就是说，这个点没有产生任何花费，，，怎么说呢= =

有点说不清楚，但是观察一下可以发现一个联通块（把所有边看成无向边）的答案都是1。

简单来说……就是个外向树啊= =。

所以……直接将i,x[i]合并起来= =然后统计联通块有几个就好了。

统计联通块有几个，等价于统计fa[i]=i的个数。

tarjan的代码：

#include<bits/stdc++.h>
using namespace std;
const int 
	N=1000005;
int n,ans,Ecnt,Time,top,sccnum;
int x[N],DFN[N],LOW[N],stk[N];
int scc[N],sz[N],into[N],Q[N];
bool instack[N],vis[N];
struct Edge{
	int next,to;
}E[N];int head[N];
void add(int u,int v){
	E[++Ecnt].next=head[u];
	E[Ecnt].to=v;
	head[u]=Ecnt;
}
void tarjan(int u){
	DFN[u]=LOW[u]=++Time;
	instack[u]=1,stk[++top]=u;
	for (int i=head[u];i;i=E[i].next){
		int v=E[i].to;
		if (!DFN[v]){
			tarjan(v);
			LOW[u]=min(LOW[u],LOW[v]);
		} else
		if (instack[v]) LOW[u]=min(LOW[u],DFN[v]);
	}
	if (DFN[u]==LOW[u]){
		sz[++sccnum]=1;
		while (stk[top]!=u){
			instack[stk[top]]=0,
			scc[stk[top--]]=sccnum;
			sz[sccnum]++;
		}
		instack[stk[top]]=0,
		scc[stk[top--]]=sccnum;
	}
}
void rebuild(){
	memset(head,0,sizeof(head));
	Ecnt=0;
	for (int i=1;i<=n;i++)
		if (scc[i]!=scc[x[i]])
			add(scc[x[i]],scc[i]),into[scc[i]]++;
}
void topo_solve(){
	int h=0,t=0;
	for (int i=1;i<=sccnum;i++)
		if (!into[i]) Q[t++]=i,vis[i]=1;
	while (h<t){
		int u=Q[h++];
		if (sz[u]>1) ans++;
		vis[u]=1;
		for (int i=head[u];i;i=E[i].next){
			int v=E[i].to;
			into[v]--;
			if (!vis[v] && !into[v]) Q[t++]=v;
		}
	}
}
int main(){
	scanf("%d",&n);
	ans=Ecnt=0;
	for (int i=1;i<=n;i++){
		scanf("%d",&x[i]);
		if (i!=x[i]) add(i,x[i]);
			else ans++;
	}
	Time=top=sccnum=0;
	for (int i=1;i<=n;i++)
		if (!DFN[i]) tarjan(i);
	rebuild();
	topo_solve();
	printf("%d\n",ans);
	return 0;
}

正解：

#include<bits/stdc++.h>
using namespace std;
int n,fa[1000005];
int getfa(int x){
	if (fa[x]!=x) fa[x]=getfa(fa[x]);
	return fa[x];
}
int main(){
	scanf("%d",&n);int x;
	for (int i=1;i<=n;i++) fa[i]=i;
	for (int i=1;i<=n;i++){
		scanf("%d",&x);
		int t1=getfa(i),t2=getfa(x);
		if (t1!=t2) fa[t2]=t1;
	}
	int ans=0;
	for (int i=1;i<=n;i++)
		if (fa[i]==i) ans++;
	printf("%d\n",ans); 
	return 0;
}