BZOJ1529/POI2005 Piggy banks

根据图中的信息,可以构造出一棵基环外向树.每个节点指向存放自己钥匙的节点.对于每一个连通块,其中必定会出现一个环,只要将环上任意一个存钱罐打破,那么这整个联通块中的所有存钱罐都能打开了.
因此答案就是构图后联通块的个数.

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
using namespace std;
const int M=1e6+5;
int fa[M],n,ans=0;
inline void rd(int &res){
    res=0;char c;
    while(c=getchar(),c<48);
    do res=(res<<1)+(res<<3)+(c^48);
    while(c=getchar(),c>=48);
}
int get(int x){
    if(fa[x]!=x)fa[x]=get(fa[x]);
    return fa[x];
}
int main(){
    int i,j,k;
    rd(n);
    for(i=1;i<=n;i++)fa[i]=i;
    for(i=1;i<=n;i++){
        rd(k);
        int x=get(k);
        if(x==i)ans++;
        else fa[i]=x;
    }
    printf("%d\n",ans);
    return 0;
}