题解 | 扫雷

import sys
n,m = map(int,input().split())
a = [li for li in sys.stdin]
total = [['*']*m for _ in range(n)]
for i in range(n):
    for j in range(m):
        count1,count2,count3=-1,-1,-1
        if a[i][j]!='*':
            if i == 0:
                count1 = 0
            elif i == n-1:
                count3 = 0
            if j == 0 :
                if count1 == -1:
                    count1 = a[i-1].count('*',j,j+2)
                count2 = a[i].count('*',j,j+2)
                if count3 == -1:
                    count3 = a[i+1].count('*',j,j+2)
            if count1==-1:
                count1 = a[i-1].count('*',j-1,j+2)
            if count2==-1:
                count2 = a[i].count('*',j-1,j+2)
            if count3==-1:
                count3 = a[i+1].count('*',j-1,j+2)
            total[i][j] = str(count1 + count2 +count3)
    print(''.join(total[i]))