蓝桥杯模拟赛第五场 - I. 引爆炸弹

题目：引爆炸弹

在一个 n×m 的方格地图上，某些方格上放置着炸弹。手动引爆一个炸弹以后，炸弹会把炸弹所在的行和列上的所有炸弹引爆，被引爆的炸弹又能引爆其他炸弹，这样连锁下去。现在为了引爆地图上的所有炸弹，需要手动引爆其中一些炸弹，为了把危险程度降到最低，请算出最少手动引爆多少个炸弹可以把地图上的所有炸弹引爆。

输入格式
第一行输两个整数 n,m，用空格隔开。
接下来 n 行，每行输入一个长度为 m 的字符串，表示地图信息。0表示没有炸弹，1表示炸弹。

数据约定：
对于 60% 的数据：1≤n,m≤100；
对于 100% 的数据： 1≤n,m≤1000；
数据量比较大，不建议用cin输入。

输出格式
输出一个整数，表示最少需要手动引爆的炸弹数。

样例输入
5 5
00010
00010
01001
10001
01000

样例输出
2

思路：利用并查集，最后的集合数就是最少要引爆的炸弹的数量

代码：

import java.util.ArrayList;
import java.util.HashSet;
import java.util.Scanner;

public class Main {
	
	private static ArrayList<P> list=new ArrayList<P>();
	private static int[] f;
	private static int n,m;

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		
		Scanner scn=new Scanner(System.in);		
		n=scn.nextInt();
		m=scn.nextInt();
		String s="";
		for(int i=0;i<n;i++){
			s=scn.next();
			for(int j=0;j<m;j++){
				char c=s.charAt(j);
				if(c=='1'){
					list.add(new P(i, j+n));
				}
			}
		}	
		scn.close();

		f=new int[n+m];
		for(int i=0;i<f.length;i++){
			f[i]=i;
		}
		
		for(int i=0;i<list.size();i++){
			P p=list.get(i);
			int f1=find(p.x);
			int f2=find(p.y);
			if(f1!=f2){
				f[f1]=f2;
			}
		}
		
		HashSet<Integer> rootSet=new HashSet<Integer>();
		for(int i=0;i<list.size();i++){
			P p=list.get(i);
			rootSet.add(find(p.x));
			rootSet.add(find(p.y));
		}
		
		System.out.println(rootSet.size());

	}
	
	private static int find(int x){
		if(x!=f[x]){
			f[x]=find(f[x]);
		}
		return f[x];
	}
	
}

class P{
	public int x;
	public int y;
	public P() {
		super();
	}
	public P(int x, int y) {
		super();
		this.x = x;
		this.y = y;
	}	
}