蓝桥杯-第八届决赛—— 合根植物

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

其合根情况参考图[p1.png]

资源约定：
峰值内存消耗 < 256M
CPU消耗  < 2000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意：主类的名字必须是：Main，否则按无效代码处理。

思路：先声明一个数组用来保存植物的编号，然后保存是否互相合根，

接下来从1开始直到最后一个一个一个遍历开始往下搜寻，把连起来的标记为已访问，

然后返回这是一整个合根植物，不然如果已被访问，则不计。最后输出就是结果一共有多少根。

第一段代码（这种方法只通过两组数据）。

下面第二段代码，用的并查集，通过了100%的数据。

（建议使用并查集）

import java.util.Scanner;

//通过两组数据，因为内存消耗太大

public class Main {

	public static int [][] a;//图
	public static boolean [] b;//标记是否被访问
	public static int m,n,k,count=0;
	public static int [][] localtion = {{0,1},{0,-1},{1,0},{-1,0}};
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		m = in.nextInt();
		n = in.nextInt();
		k = in.nextInt();
		a = new int [m*n+1][m*n+1];
		b = new boolean [m*n+1];
		for(int i=0;i<k;i++){
			int p = in.nextInt();
			int l = in.nextInt();
			a[p][l] = 1;
			a[l][p] = 1;
		}
		for(int i=0;i<m;i++){
			for(int j=0;j<n;j++){
				a[i*n+j+1][i*n+j+1] = 1;
				if(dfs(i,j,i*n+j+1)){
					count++;
				}
			}
		}
		System.out.println(count);
	}
	
	private static boolean dfs(int i,int j,int c) {

		if(i<0||j<0||i>=m||j>=n){//判断是否越界
			return false;
		}
		int temp = i*n+j+1; 
		if(b[temp]){//判断是否被访问过
			return false;
		}
		
		if(a[temp][c]==1){
			b[temp] = true;
			for(int q = 0;q<4;q++){
				dfs(i+localtion[q][0],j+localtion[q][1],temp);
			}
		}else{
			return true;
		}
		return true;
	}
}

 

import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
//并查集
public class Main {

	public static int [] a;
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		int m = in.nextInt();
		int n = in.nextInt();
		int k = in.nextInt();
		a = new int [n*m+1];
		for(int i=1;i<n*m+1;i++){
			a[i] = i;
		}
		for(int i=0;i<k;i++){
			int p = in.nextInt();
			int l = in.nextInt();
			show(p,l);
		}
		Set<Integer> set = new HashSet<Integer>();
		for(int i=1;i<n*m+1;i++){
			set.add(dfs(i));
		}
		System.out.println(set.size());
	}
	public static void show(int p,int l){
		if(dfs(p)==dfs(l)){
			return;
		}
		a[dfs(p)] = dfs(l);
		
	}
	public static int dfs(int i){
		if(a[i]==i){
			return i;
		}
		return a[i] = dfs(a[i]);
	}
	
}