2019藍橋杯JAVA練習 合根植物(並查集+路徑壓縮)
阿新 • • 發佈:2018-11-28
問題描述
w星球的一個種植園,被分成 m * n 個小格子(東西方向m行,南北方向n列)。每個格子裡種了一株合根植物。
這種植物有個特點,它的根可能會沿著南北或東西方向伸展,從而與另一個格子的植物合成為一體。
如果我們告訴你哪些小格子間出現了連根現象,你能說出這個園中一共有多少株合根植物嗎?輸入格式
第一行,兩個整數m,n,用空格分開,表示格子的行數、列數(1<m,n<1000)。
接下來一行,一個整數k,表示下面還有k行資料(0<k<100000)
接下來k行,第行兩個整數a,b,表示編號為a的小格子和編號為b的小格子合根了。
格子的編號一行一行,從上到下,從左到右編號。
比如:5 * 4 的小格子,編號:
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
17 18 19 20樣例輸入
5 4
16
2 3
1 5
5 9
4 8
7 8
9 10
10 11
11 12
10 14
12 16
14 18
17 18
15 19
19 20
9 13
13 17樣例輸出
5
樣例說明
其合根情況參考下圖
題目大意:
求這張圖的連通塊數量。
例子中每一塊連通的紅線為一塊,編號6為單獨一塊連通塊。
import java.util.Scanner; public class Main { static int father[] = null; public static void main(String[] args) { Scanner s = new Scanner(System.in); int m = s.nextInt(); int n = s.nextInt(); int k = s.nextInt(); father = new int[m*n+1]; for(int i=1;i<=m*n;i++){ father[i] = i; } for(int i=0;i<k;i++){ int a = s.nextInt(); int b = s.nextInt(); join(a,b); } int cnt = 0; for(int i=1;i<=m*n;i++){ if(father[i]==i){ cnt++; } } System.out.println(cnt); } public static void join(int a,int b){ if(getTop(a) != getTop(b)){ father[getTop(a)]= getTop(b); } } public static int getTop(int a){ if(father[a]==a){ return a; }else{ father[a] = getTop(father[a]); return father[a]; } } }