1. 程式人生 > >2019藍橋杯JAVA練習 合根植物(並查集+路徑壓縮)

2019藍橋杯JAVA練習 合根植物(並查集+路徑壓縮)

問題描述

  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];
		}
	}

}