问题 B: 沈老师的岛

问题 B: 沈老师的岛

题目描述

沈老师天天说象山是个好地方，鹤浦更是个好地方。由于鹤浦是一个岛屿，沈老师更是有一个外号叫做“岛主”。现在“岛主”来请你帮帮忙，他想知道，他的家乡附近有多少个独立的岛屿？
给定一个由 ‘@’（陆地）和 '’（水）组成的的二维网格，计算独立的岛屿的数量。一个岛被水包围，并且它是通过水平或垂直8个方向上相邻的陆地连接而成的。你可以假设网格的四个边均被水包围。
输入
样例输入由多组测试数据组成。第一行输入两个正整数n和m分别代表网格的高和宽 ( 0 < n,m <= 100 )
接下来输入一个nm的网格，网格内只由字符 ‘@’ 和 ‘*’ 组成，@代表陆地，*代表海洋。

输出

输出独立的岛屿的数量

样例输入 Copy

1 1
*
3 5
*@*@*
**@**
*@*@*
1 8
@@****@*

样例输出 Copy

0
1
2

思路
这题可以使用并查集和dfs，下面是我用并查集去完成的。

AC代码

#include<bits/stdc++.h>
using namespace std;
int s[10005];
//int dir[8][2] = { {0,1},{1,0},{0,-1},{-1,0},{1,1},{1,-1},{-1,1},{1,-1} };
char mp[105][105];
int ans = 0;
void init(int n){
	for(int i = 0 ; i < n ; i ++){
		s[i] = i;
	}
}
/*void inits(int n){
	for(int i = 0 ; i < n ; i ++){
		cout << s[i] << " ";
	}
}*/
int find(int x){
	if(s[x] == x){
		return x;
	}
	s[x] = find(s[x]);
	return s[x];
}
int marge(int x,int y){
	int tx = find(x);
	int ty = find(y);
	if(tx != ty){
		s[tx] = ty;
		ans --;
	}
}
int main(){
	int n,m;
	while(cin >> n >> m){
		ans = 0;
		memset(mp,'\0',sizeof(mp));
		for(int i = 0 ; i < n ; i ++){
				scanf("%s",&mp[i]);
		}
		for(int i = 0 ; i < n ; i ++){
			for(int j = 0 ; j < m ; j ++){
				if(mp[i][j] == '@') ans ++; 
			}
		}
		/*for(int i = 1 ; i <= n ; i ++){
			for(int j = 0 ; j <= m ; j ++){
				cout << mp[i][j] << " ";			//输入判断是否正确 
			}
			cout << endl;
		}*/
		init(n * m);
		for(int i = 0 ; i < n ; i ++){
			for(int j = 0 ; j < m ; j ++){
				if(mp[i][j] == '@' && mp[i - 1][j] == '@' && i - 1 >= 0){
					marge(m * i + j,(i - 1) * m + j);	
				}
				if(mp[i][j] == '@' && mp[i + 1][j] == '@' && i + 1 < n ){
					marge(m * i + j,(i + 1) * m + j);	
				}
				if(mp[i][j] == '@' && mp[i][j - 1] == '@' && j - 1 >= 0){
					marge(m * i + j,i * m + j - 1);	
				}
				if(mp[i][j] == '@' && mp[i][j + 1] == '@' && j + 1 < m){
					marge(m * i + j,i * m + j + 1);	
				}
				if(mp[i][j] == '@' && mp[i + 1][j + 1] == '@' && i + 1 < n && j + 1 < m){
					marge(m * i + j,(i + 1) * m + j + 1);
				}
				if(mp[i][j] == '@' && mp[i + 1][j - 1] == '@' && i + 1 < n && j - 1 >= 0){
					marge(m * i + j,(i + 1) * m + j - 1);
				}
				if(mp[i][j] == '@' && mp[i - 1][j + 1] == '@' && i - 1 >= 0 && j + 1 < m){
					marge(m * i + j,(i - 1) * m + j + 1);
				}
				if(mp[i][j] == '@' && mp[i - 1][j - 1] == '@' && i - 1 >= 0 && j - 1 >= 0){
					marge(m * i + j,(i - 1) * m + j - 1);
				}
			}
		}
	//	inits(n * m);
		cout << ans << endl;
	}
}