学习并查集戳这里
问题描述
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
样例说明
其合根情况参考下图
思路:
纯并查集问题,直接上代码 ,不过要注意让求的是最后还有多少合根植物,单独一个也是一个合根植物(eg:6)!
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class 合根植物 {
private int[] parent;//存放父节点
private int size;//数组长度
private int ans;//结果
//初始化
public void unionFind(int size){
this.size=size;
parent=new int[size];
for(int i=1;i<size;i++){
parent[i]=i;
}
}
//find函数(找大哥)
public int find(int element){
while(element!=parent[element]){
element=parent[element];//向最终父节点方向搜索
}
return element;
}
//判断是否连通,如果最终的父结点是相同的就是连通的
// public Boolean isConnected(int firstElement,int secondElement){
// return find(firstElement)==find(secondElement);
// }
//合并
public void unionElement(int firstElement,int secondElement){
int first=find(firstElement);
int second=find(secondElement);
if(first!=second){
parent[first]=second;
ans--;//两个合成一个,合根植物数量减一
}
}
public static void main(String[] args) throws IOException {
// TODO 自动生成的方法存根
BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
String matrix[]=br.readLine().split(" ");
int m,n;
m=Integer.parseInt(matrix[0]);
n=Integer.parseInt(matrix[1]);
合根植物 union=new 合根植物();
union.unionFind(m*n+1);//初始化
union.ans=m*n;//刚开始每个都是自己的最终父节点,即有m*n个合根植物
String K=br.readLine();
int k=Integer.parseInt(K);//合并次数
for(int i=0;i<k;i++){
//合根(放到一个集合中)
String tmp[]=br.readLine().split(" ");
int a=Integer.parseInt(tmp[0]);
int b=Integer.parseInt(tmp[1]);
union.unionElement(a, b);
}
System.out.println(union.ans);
}
}