L2-4 寻宝图 天梯赛题补

L2-4 寻宝图

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

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

思路就是爆搜,如果>='1’就表明是岛，直接把岛逛一遍，做上标记下次不来这个岛了，每进一次岛，num++，全局变量flag每次搜之前设置为0，搜的时候发现宝藏了就flag=1，宝藏++，

代码

#include <vector>
using namespace std;
const int N = 1e5+100;
char a[N];
bool st[N];
int n,m;
int get(int x,int y){
    return x*m+y;
}
bool flag = 1;
int dx[]={0,0,1,-1};
int dy[]={1,-1,0,0};
void dfs(int x,int y){
    int u=get(x,y);
    st[u]=1;
    if(a[u]>'1') flag=1;
    for(int i=0;i<4;i++){
        int a1=x+dx[i],b1=y+dy[i];
        u=get(a1,b1);
        if(a1>=0&&a1<n&&b1>=0&&b1<m&&!st[u]&&a[u]>='1') dfs(a1,b1);
    }
}
int main(){

    cin>>n>>m;
	getchar();
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            char c;
            c=getchar();
            a[get(i,j)]=c;
        }
        getchar();
    }
    int num=0,num1=0;
    for(int i=0;i<n;i++)
        for(int j=0;j<m;j++){
            flag = 0;
            if(!st[get(i,j)]&&a[get(i,j)]>='1') {
                num++;
                dfs(i,j);
                if(flag) num1++;
            }
        }
    
    cout<<num<<" "<<num1;
}