合根植物(并查集)

学习并查集戳这里

问题描述

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

}
发布了81 篇原创文章 · 获赞 91 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44593822/article/details/103114209
今日推荐