描述
在某城市里住着n个人,任何两个认识的人不是朋友就是敌人,而且满足:
1、我朋友的朋友是我的朋友;
2、我敌人的敌人是我的朋友;
所有是朋友的人组成一个团伙。告诉你关于这n个人的m条信息,即某两个人是朋友,或者某两个人是敌人,请你编写一个程序,计算出这个城市最多可能有多少个团伙?
输入
第1行为n和m,1<n<1000,1≤m≤100 000;
以下m行,每行为p x y,p的值为0或1,p为0时,表示x和y是朋友,p为1时,表示x和y是敌人。
输出
一个整数,表示这n个人最多可能有几个团伙。
思路
先将p为0是的所有关系都加入并查集,将p为1的关系塞入vector二维数组,遍历一遍之后将p为一且共有一个敌人的关系两两加入并查集
代码如下(示例):
#include <bits/stdc++.h>
using namespace std;
string str;
int a[110000];
int js=1;
int f(int x)
{
if(a[x]==x)
return x;
else
return a[x]=f(a[x]);
}
map<string,int>mp;
map<int,string>mp1;
int main()
{
for(int i=0;i<=100000;i++)
a[i]=i;
int aa,bb;
while(cin>>str,str!="$")
{
if(str[0]!='?')
{
if(str[0]=='#')
{
str.erase(str.begin());
if(mp.find(str)==mp.end())
{
mp[str]=js;
mp1[js++]=str;
}
aa=f(a[mp[str]]);
}
else
{
str.erase(str.begin());
if(mp.find(str)==mp.end())
{
mp[str]=js;
mp1[js++]=str;
}
bb=f(a[mp[str]]);
if(bb!=aa)
a[bb]=aa;
}
}
else
{
str.erase(str.begin());
cout<<str<<" ";
int zz=f(a[mp[str]]);
cout<<mp1[zz]<<endl;
}
}
}
总结
代码有点小乱有些知识可以简化一下