并查集

来源:互联网 发布:mac os 10.12 beta8 编辑:程序博客网 时间:2024/06/06 17:14

并查集:对要求合并的集合,先查找他们是否有亲戚(自己集合里面的某个元素也在其他集合里面)在其他集合,如果有那么就合并到一个更大的集合里面

//  [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];elsereturn 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;elsefor(int i=0;i<NODE_NUM;++i)if(arr[i]==arr[p])arr[i] = arr[q];--count_set;return true;}elsereturn false;}