PTA L2-048 寻宝图

题目链接

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

输入格式：

输入第一行给出 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

解题思路

很经典的连通块问题，dfs和bfs都可以，这里我使用dfs。 每次遇到岛屿或宝藏就深搜，并且搜这个点上下左右的四个点，只要不是0（水域），就将它变成0。最后这一整块连通的岛屿就都变成了水1。在搜的过程中，如果碰到了宝藏，就要标记一下，并统计宝藏的个数即可。

AC代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<vector> 
#include<map>
#include<queue>
#include<cmath>
using namespace std;
#define int long long
#define endl '\n'
typedef pair<int, int> PII;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;

const int N = 100005;
int n, m;
string g[N];
int dx[4] = { 0,1,0,-1 }, dy[4] = { 1,0,-1,0 };
int cnt1 = 0, cnt2 = 0;
bool f = false;

void dfs(int x,int y)
{
    if (x < 0 || x >= n || y < 0 || y >= m || g[x][y] == '0') return;
    if (g[x][y] > '1') f = true;
    g[x][y] = '0';
    for (int i = 0; i < 4; i++)
    {
        dfs(x + dx[i], y + dy[i]);
    }
}

signed main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m;
    for (int i = 0; i < n; i++)  cin >> g[i];


    for (int i = 0; i < n; i++)
        for (int j = 0; j < m; j++)
            if (g[i][j] != '0')
            {
                f = false;
                cnt1++;
                dfs(i, j);
                if (f) cnt2++;
            }
    cout << cnt1 << " " << cnt2;
    return 0;
}