/***************************************
*@time 2016/08/14 09:09
*@palce DHU.5005
*description 並查集操作
**************************************/
#include<cstdio>
#include<map>
using namespace std;
/*********************************************************************
*@函數名 find_Root
*@parameter map<int,int>* Union 儲存並查集,如果節點是非根節點則Union[node]表示node的前驅節點,
* 如果節點是根節點則Union[node]表示以node爲跟節點的樹的節點數
*@parameter map<int,bool>* Root 判斷節點是否爲根節點
*@parameter int x 要找到根節點的點
*@description 根據Union,Root找到一個節點的根節點
**************************************************************************************************************/
int find_Root(map<int,int>* Union,map<int,bool>* Root,int x);
/*********************************************************************
*@函數名 creat_Tree
*@parameter map<int,int>* Union 儲存並查集,如果節點是非根節點則Union[node]表示node的前驅節點,
* 如果節點是根節點則Union[node]表示以node爲跟節點的樹的節點數
*@parameter map<int,bool>* Root 判斷節點是否爲根節點
*@parameter int x ,int y 路徑的兩端節點
*@description 將節點加入並查集
**************************************************************************************************************/
void creat_Tree(map<int,int>* Union,map<int,bool>* Root,int x,int y);
/*********************************************************************
*@函數名 count_Trees_Num
*@parameter map<int,bool>* Root 判斷節點是否爲根節點
*@return value int 樹的個數,也就是根節點的個數
*@description 統計樹的個數,也就是根節點的個數
**************************************************************************************************************/
int count_Trees_Num(map<int,bool>* Root);
int find_Root(map<int,int>* Union,map<int,bool>* Root,int x)
{
while(!(*Root)[x]) x=(*Union)[x];
return x;
}
void creat_Tree(map<int,int>* Union,map<int,bool>* Root,int x,int y)
{
int rootX=find_Root(Union,Root,x);
int rootY=find_Root(Union,Root,y);
//x,y不在同一顆樹中,將x所在的樹與y所在的是合併
if(rootX!=rootY)
{
if((*Union)[rootX]>(*Union)[rootY])//x樹較大,則將y數加入x數
{
(*Union)[rootX]+=(*Union)[rootY];//統計合併後樹中節點的個數
(*Union)[rootY]=rootX;
(*Root)[rootY]=false;
}
else
{
(*Union)[rootY]+=(*Union)[rootX];
(*Union)[rootX]=rootY;
(*Root)[rootX]=false;
}
}
}
int count_Trees_Num(map<int,bool>* Root)
{
map<int,bool>::iterator it;
int sum=0;
for(it=(*Root).begin();it!=(*Root).end();it++)
{
if(it->second==true) sum++;
}
return sum;
}
int main()
{
int num_Nodes;
int num_Routes;
map<int,int> Union;
map<int,bool> Root;
while(scanf("%d",&num_Nodes)&&num_Nodes)
{
Union.clear();
Root.clear();
for(int i=0;i<num_Nodes;i++)
{
int node;
scanf("%d",&node);
Root[node]=true;
Union[node]=node;
}
scanf("%d",&num_Routes);
for(int i=0;i<num_Routes;i++)
{
int start_Node;
int end_Node;
scanf("%d %d",&start_Node,&end_Node);
creat_Tree(&Union,&Root,start_Node,end_Node);
}
int sum=count_Trees_Num(&Root);
printf("sum=%d\n",sum-1);
}
}
測試樣例:
http://acm.hdu.edu.cn/showproblem.php?pid=1232