计蒜客 A1139 ----引爆炸弹 【并查集 or dfs】

引爆炸弹

思路1 ：dfs

dfs，先点爆炸弹，然后把该炸弹行和列标记了，继续枚举改行的其他炸弹(引爆)，一直dfs下去(一直引爆)，直到不能引爆。引爆的炸弹和点爆的炸弹都变为0，dfs的次数就是点爆的次数。时间复杂度为O(n*m)

代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<map>
#include<queue>
#include<string>
#include<vector>
#include<utility>
#include<algorithm>
using namespace std;
const int MAXN=1e3+5;
char mp[MAXN][MAXN];
bool row[MAXN],line[MAXN];
int n,m;
inline int read(){
	int s=0,w=1;
	char ch=getchar();
	for(;ch>'9' || ch<'0';ch=getchar()){
		if(ch=='-')	w=-1;
	}
	for(;ch<='9' && ch>='0';ch=getchar()){
		s=(s<<3)+(s<<1)+(ch^'0');
	}
	return s*w;
}
void dfs(int x,int y){
	mp[x][y]='0';
	if(!row[x]){//未被引爆的行 
		row[x]=true;
		for(int i=0;i<m;i++)//引爆该行的所有炸弹 
			if(mp[x][i]=='1')
				dfs(x,i);
	}
	if(!line[y]){
		line[y]=true;//未被引爆的列 
		for(int i=0;i<n;i++){//引爆该列的所有炸弹 
			if(mp[i][y]=='1')
				dfs(i,y);
		}
	}
}
int main(){
	char ch;
	n=read();m=read();
	for(int i=0;i<n;i++)
		scanf("%s",&mp[i]);
	int ans=0;
	for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
			if(mp[i][j]=='1')	dfs(i,j),ans++;//点爆炸弹 
	printf("%d\n",ans);
	return 0;
}

思路2 ：并查集

将炸弹的行和列合并为同一个集合，遍历炸弹，如果行没走过，直接点爆该行的根结点，即这个集合所有都引爆。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 1e3 + 5;
int fa[maxn*2];
char s[maxn][maxn];
bool vis[maxn*2];
void init(int n){
	for (int i = 0; i < n; i++){
		fa[i] = i;
	}
}
int find(int x){
	return fa[x] == x ? fa[x] : fa[x] = find(fa[x]);
}
void _union(int x, int y){
	x = find(x);
	y = find(y);
	if (x != y)
		fa[x] = y; 
}
void solve(){
	int n, m;
	scanf("%d%d", &n, &m);
	init(n+m);
	for (int i = 0; i < n; ++i){
		scanf("%s",&s[i]);
		for (int j = 0; j < m; ++j){
			if (s[i][j] == '1'){
				_union(i, j + n);
				vis[i] = vis[j + n] = true;
			}
		}
	}
	int ans = 0;
	for (int i = 0; i < n + m; ++i){
		if(!vis[i])	continue;
		if (find(i) == i)	ans++;
	}
	printf("%d\n",ans);
}
int main(){
	solve();
	return 0;
}