并查集:对要求合并的集合,先查找他们是否有亲戚(自己集合里面的某个元素也在其他集合里面)在其他集合,如果有那么就合并到一个更大的集合里面
// [9/30/2013 qingezha] Union-Find 并查集
// 初始化所有的结点用整数表示,比如(0--n-1),在处理输入pair之前,他们分属于不同的组,每个结点都是孤立的
// 可以用数组表示这种关系,数组的index表示结点,而对应的值表示组号
int const NODE_NUM = 1001;
int count_set = NODE_NUM;
int arr[NODE_NUM]={0};
void UF_init() //初始化
{
for (int i=0;i<NODE_NUM;++i)
arr[i] = i;
}
int find(int p) //查找这个结点属于 哪个组
{
if(p>=0&&p<NODE_NUM)
return arr[p];
else
return -1;
}
bool connect(int p,int q) //判断是否连接
{
if(p>=0 && p<NODE_NUM && q>=0 && q<NODE_NUM)
return arr[p]==arr[q];
else
return false;
}
int count() //组的个数
{
return count_set;
}
bool union_two(int p,int q) //合并2个,使之为一个组
{
if(p>=0 && p<NODE_NUM && q>=0 && q<NODE_NUM)
{
if(arr[p]==arr[q])
return true;
else
for(int i=0;i<NODE_NUM;++i)
if(arr[i]==arr[p])
arr[i] = arr[q];
--count_set;
return true;
}
else
return false;
}