1. 程式人生 > >HDU1272: 小希的迷宮(基礎並查集)

HDU1272: 小希的迷宮(基礎並查集)

題目描述:
上次Gardon的迷宮城堡小希玩了很久(見Problem B),現在她也想設計一個迷宮讓Gardon來走。但是她設計迷宮的思路不一樣,首先她認為所有的通道都應該是雙向連通的,就是說如果有一個通道連通了房間A和B,那麼既可以通過它從房間A走到房間B,也可以通過它從房間B走到房間A,為了提高難度,小希希望任意兩個房間有且僅有一條路徑可以相通(除非走了回頭路)。小希現在把她的設計圖給你,讓你幫忙判斷她的設計圖是否符合她的設計思路。比如下面的例子,前兩個是符合條件的,但是最後一個卻有兩種方法從5到達8。

輸入描述:
輸入包含多組資料,每組資料是一個以0 0結尾的整數對列表,表示了一條通道連線的兩個房間的編號。房間的編號至少為1,且不超過100000。每兩組資料之間有一個空行。 整個檔案以兩個-1結尾。

輸出描述:
對於輸入的每一組資料,輸出僅包括一行。如果該迷宮符合小希的思路,那麼輸出”Yes”,否則輸出”No”。

Sample Input
6 8 5 3 5 2 6 4
5 6 0 0

8 1 7 3 6 2 8 9 7 5
7 4 7 8 7 6 0 0

3 8 6 8 6 4
5 3 5 6 5 2 0 0

-1 -1

Sample Output
Yes
Yes
No

     很顯然,題目就是判斷設計圖是不是連通無環的圖。

     所以可以採用並查集,如果新讀入的兩個元素在同一個集合裡,顯然構成環,say no

     至於判斷是否連通,就查詢看是否只有一個根結點。

#include <stdio.h>
#include <string.h>
int g[100000];

int find(int x)
{
    int r=x;
    while(r!=g[r])
        r=g[r];
    return r;
}
void merge(int x,int y)
{
    int xx=find(x);
    int yy=find(y);
    if(x!=y)
       g[xx]=yy;
}

int main()
{
    int x,y;
    bool flag=true;
    memset(g,0,sizeof(g));
    while(scanf("%d %d",&x,&y)==2)
    {
        if(x==-1&&y==-1) return 0;
        if(x==0&&y==0)                   //輸入0 0,重置
        {
            if(!flag)   printf("No\n");
            else printf("Yes\n");
            memset(g,0,sizeof(g));
            continue;
        }
        if(g[x]==0)  g[x]=x;                //為0,代表沒有初始化和改變過,所以進行初始化
        if(g[y]==0) g[y]=y;
        if(find(x)==find(y)) { flag=false;}
        else merge(x,y);
    }
}