L2-048 寻宝图(25分)

题目描述

给定一幅地图，其中有水域，有陆地。被水域完全环绕的陆地是岛屿。有些岛屿上埋藏有宝藏，这些有宝藏的点也被标记出来了。本题就请你统计一下，给定的地图上一共有多少岛屿，其中有多少是有宝藏的岛屿。

输入格式：
输入第一行给出 2 个正整数 N 和 M（1<N×M≤10^5），是地图的尺寸，表示地图由 N 行 M 列格子构成。随后 N 行，每行给出 M 位个位数，其中 0 表示水域，1 表示陆地，2-9 表示宝藏。
注意：两个格子共享一条边时，才是“相邻”的。宝藏都埋在陆地上。默认地图外围全是水域。

输出格式：
在一行中输出 2 个整数，分别是岛屿的总数量和有宝藏的岛屿的数量。

输入样例：

10 11
01000000151
11000000111
00110000811
00110100010
00000000000
00000111000
00114111000
00110010000
00019000010
00120000001

输出样例：

7 2

代码实现

#include <bits/stdc++.h>
using namespace std;
//const int maxn = 10005;
//string grid[maxn];
//bool st[maxn][maxn];
const int dy[] = {0,1,0,-1};
const int dx[] = {1,0,-1,0};
int N,M;
string *grid; // 保存地图 
bool **st;	// 标记是否访问该点 
int cnt, ans; // 岛屿数量 有宝藏的岛屿数量 
void init() {
	// 申请内存
	st = new bool*[N];
	for(int i=0; i < N; ++i) st[i] = new bool[M];
	grid = new string[N];
}
void destroy() {
	// 释放内存 
	for(int i=0; i < N; ++i) delete []st[i];
	delete []st;
	delete []grid;
}
bool judge(int idx,int K) {
	if(idx >= 0 && idx < K) return true;
	return false;
}
bool dfs(int x, int y) {
	bool ret = false; // 该岛屿是否有宝藏 
	if(grid[x][y] >= '2' && grid[x][y] <= '9') 
		ret = true;
	st[x][y] = true;
	for(int i=0; i < 4; ++i) {
		int nx = x + dx[i];
		int ny = y + dy[i];
		if(judge(nx, N) && judge(ny, M) && st[nx][ny] == false && grid[nx][ny] != '0') {
			ret |= dfs(nx, ny);
		}
	} 
	return ret;
}
int main() {
    cin >> N >> M;
    init();
	for(int i=0; i < N; ++i) {
		cin >> grid[i];
	} 
	for(int i=0; i < N; ++i) {
		for(int j=0; j < M; ++j) {
			if(!st[i][j] && grid[i][j] != '0') {
				// 遇到陆地，搜索所有与其相连的陆地 
				++cnt;
				if(dfs(i, j)) ++ans;
			}
		}
	}
	cout << cnt << " " << ans << endl;
	destroy();
    return 0;
}