经典算法之并查集 合根植物

题目描述:

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.Scanner;

public class Main {
	static int[] tb;   //整棵树 
	static int num;   //合并数
	public static void main(String[] args) {
		Scanner input = new Scanner(System.in);
		String[] ss = input.nextLine().trim().split(" +");   //根据一个或者多个空格分割
		
		int M = Integer.parseInt(ss[0]);
		int N = Integer.parseInt(ss[1]);
		
		tb = new int[M*N + 1];  //整棵树 0不存
		int k = Integer.parseInt(input.nextLine().trim());   //有k条连通数据
		for (int i = 0; i < k; i++) {
			ss = input.nextLine().trim().split(" +");
			
			//ss[0] ss[1]  表示ss[0]和ss[1]连
			fun(Integer.parseInt(ss[0]), Integer.parseInt(ss[1]));
		}
		
			System.out.println(M*N - num);   //总数减掉有合并数便为合根植物的根数
	}
	
	//a与b连通
	public static void fun(int a, int b) {

		int ra = find(a);   //查询a的父节点 用祖先的号代表整个家族
		int rb = find(b);
		
		if (ra == rb) return;   //若两树的根节点相同则表示本身就为一颗合并数
		
		union(ra, rb);  //合并ra,rb
		num++;
	}
	public static void union(int ra, int rb) {

		tb[ra] = rb;
	}
	public static int find(int x) {   //查找其父节点

		if(tb[x] == 0) return x;
		return tb[x] = find(tb[x]);  //并查集
	}
}

运行调试:

github代码:https://github.com/striner/javaCode/blob/master/Co-rooted%20plants