合根植物-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

参考解答：

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	static int[] tree;
	static boolean[] arr;
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int num_line = sc.nextInt();
		int num_column = sc.nextInt();
		tree = new int[num_line * num_column + 1];
		init(num_line * num_column);
		int k = sc.nextInt();
		arr = new boolean[num_line * num_column + 1];
		Arrays.fill(arr, 1, arr.length, false);
		for (int i = 0; i < k; i++) {
			int a = sc.nextInt();
			int b = sc.nextInt();
			unite(a, b);
		}
		int sum = 0;
		for (int i = 1; i <= num_line * num_column; i++) {
			arr[find_root(i)] = true;
		}
		for (int i = 1; i <= num_line * num_column; i++) {
			if (arr[i] == true) {
				sum++;
			}
		}
		System.out.println(sum);
	}
	
	private static void init(int n) {
		for (int i = 1; i <= n; i++) {
			tree[i] = i;
		}
	}
	
	private static int find_root(int node) {
		if (tree[node] == node) {
			return node;
		}
		else {
			return tree[node] = find_root(tree[node]);
		}
	}
	
	private static void unite(int node_a, int node_b) {
		int root_a = find_root(node_a);
		int root_b = find_root(node_b);
		if (root_a != root_b) {
			tree[root_a] = root_b;
		}
	}
}