L2-4 寻宝图(Python3)

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

输入格式：

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

import sys
sys.setrecursionlimit(10**9)
n,m = map(int,input().split())
arr = [[] for i in range(100005)]
ans = 0 #表示岛屿的数量
ant = 0 #表示的是有宝藏的数量
dx = [1,0,-1,0]
dy = [0,1,0,-1]
def dfs(x:int,y:int):
    global n
    global m
    global flag
    if x < 0 or x >= n or y < 0 or y >= m or arr[x][y] == '0':
        return
    if arr[x][y] > '1':
        flag = 1
    arr[x][y] = '0'
    for i in range(4):
        dfs(x+dx[i],y+dy[i])
for i in range(n):
    arr[i] = list(input())
for i in range(n):
    for j in range(m):
        if arr[i][j] >'0':
            flag = 0
            ans += 1
            dfs(i,j)
            if flag:
                ant += 1
print(ans,ant)