来自《算法竞赛入门经典训练指南》
1.题目原文
有一些简单化合物,每个化合物有两种元素组成。你是一个装箱工人,负责把化合物装上车,如果车上存在k个简单化合物,正好包含k个元素,会发生爆炸。为了安全起见,每拿到一个化合物,如果它与已装车的化合物形成易爆炸化合物,不应该把它装上车,否则装上车。输出有多少个没有装上车的化合物。
2.解题思路
把每种元素看成顶点,则一个简单化合物对应一条边,当整个图中存在环时,组成环的边对应的化合物是危险的,反之是安全的。
可以利用并查集,高效检查每个化合物的组成元素x和y是否在一个集合中。如果是,则拒绝。反之则接受。
3.AC代码
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>
#include<cmath>
#include<bitset>
#include<sstream>
using namespace std;
#define INF 0x7fffffff
#define maxn 100005
int par[maxn];//父亲
int Rank[maxn];//树的高度
//初始化
void init(int n)
{
for(int i=0;i<n;i++){
par[i]=-1;
Rank[i]=0;
}
}
//查询父亲,路径压缩
int Find(int x)
{
int s;
for(s=x;par[s]!=-1;s=par[s]);
while(s!=x){
int temp=par[x];
par[x]=s;
x=temp;
}
return s;
}
//合并x,y属于的集合
void unite(int x,int y)
{
x=Find(x);y=Find(y);
if(x==y) return;
if(Rank[x]<Rank[y]){
par[x]=y;
}
else{
par[y]=x;
if(Rank[x]==Rank[y]) Rank[x]++;
}
}
bool same(int x,int y)
{
return Find(x)==Find(y);
}
int main()
{
int x,y;
while(scanf("%d",&x)==1){
init(maxn);
int res=0;
while(x!=-1){
scanf("%d",&y);
if(same(x,y)) res++;
else unite(x,y);
scanf("%d",&x);
}
printf("%d\n",res);
}
return 0;
}