给定一幅地图,其中有水域,有陆地。被水域完全环绕的陆地是岛屿。有些岛屿上埋藏有宝藏,这些有宝藏的点也被标记出来了。本题就请你统计一下,给定的地图上一共有多少岛屿,其中有多少是有宝藏的岛屿。
输入格式:
输入第一行给出 2 个正整数 N 和 M(1<N×M≤105),是地图的尺寸,表示地图由 N 行 M 列格子构成。随后 N 行,每行给出 M 位个位数,其中 0
表示水域,1
表示陆地,2
-9
表示宝藏。
注意:两个格子共享一条边时,才是“相邻”的。宝藏都埋在陆地上。默认地图外围全是水域。
输出格式:
在一行中输出 2 个整数,分别是岛屿的总数量和有宝藏的岛屿的数量。
输入样例:
10 11
01000000151
11000000111
00110000811
00110100010
00000000000
00000111000
00114111000
00110010000
00019000010
00120000001
输出样例:
7 2
解题思路:
由于数据较大,所以采用map<pair<int,int>>的方式建立地图,并且采用离散的方式保存陆地和宝藏所在的地点。然后就是根据每个点进行BFS。
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
map<pair<int, int>, char>mp;//地图
queue<pair<int, int>>q;//陆地点
map<pair<int, int>, int>vis;//判断是否访问过
int dx[] = { 1,0,-1,0 };
int dy[] = { 0,1,0,-1 };
int main() {
int n, m; cin >> n >> m;
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
char x; cin >> x;
mp[{i, j}] = x;
if (x != '0') {
q.push({ i,j });
}
}
}
int cnt = 0; //宝藏岛屿数量
int ans = 0; //总岛屿数量
while (!q.empty()) {//对每个点进行BFS
auto u = q.front(); q.pop();
if (vis[u])continue;
queue<pair<int, int>>nq;
bool flag = false;//判断当前岛屿是否有宝藏
nq.push(u);
if (mp[{u.first, u.second}] >= '2')flag = true;//当前点有宝藏
while (!nq.empty()) {//BFS
auto ed = nq.front(); nq.pop();
if (vis[ed])continue;
vis[ed] = 1;
for (int i = 0; i < 4; i++) {
int nx = ed.first + dx[i];
int ny = ed.second + dy[i];
if (nx<1 || nx>n || ny<1 || ny>m || mp[{nx, ny}] == '0')continue;
if (mp[{nx, ny}] >= '2') {//有宝藏
flag = true;
}
nq.push({ nx,ny });
}
}
if (flag) {
cnt++;//有宝藏的岛屿+1
}
ans++;//总岛屿+1
}
cout << ans << " " << cnt << endl;
return 0;
}