PTA L2-048 寻宝图 (25 分)

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

输入格式：

输入第一行给出 2 个正整数 $N$ 和 $M$ $（1<N\times M\le 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;

typedef pair<int,int> PII;
int n,m;
int res,ans;

void bfs(vector<vector<char>> &g, int sx, int sy)
{
    int dx[]={0,1,0,-1},dy[]={1,0,-1,0};
	queue<PII> Q;
    Q.push({sx,sy});

    res++;
    int ok=0;
    if(g[sx][sy]>'1') ok=1;
    g[sx][sy]='0';

    while(!Q.empty())
    {
        PII t=Q.front();Q.pop();
        for(int i=0;i<4;i++)
        {
            int x=t.first+dx[i],y=t.second+dy[i];
            if(x<1 || x>n || y<1 || y>m) continue;
            if(g[x][y]=='0') continue;
            if(g[x][y]>'1') ok=1;
            g[x][y]='0';
            Q.push({x,y});
        }
    }
    ans+=ok;
}

int main()
{
	cin>>n>>m;
	vector<vector<char>> g(n+1,vector<char>(m+1));
	for(int i=1;i<=n;i++) 
		for(int j=1;j<=m;j++)
			cin>>g[i][j];
	
	res=ans=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			if(g[i][j]-'0')
				bfs(g,i,j);
	cout<<res<<" "<<ans;

    return 0;
}