题目描述
解题思路
这一看就是个经典二维数组动态规划问题,这个和求路径和不一样,这个是求周围的地雷数。所以我们需要了解dp的终止条件以及初始条件判断(需要借用方向列表)
代码
n,m=map(int,input().strip().split())
maps=[[*map(int,input().strip().split())] for _ in range(n)]
dp=[[0]*m for _ in range(n)]
direct=[(1,0),(0,1),(-1,0),(0,-1),(1,1),(1,-1),(-1,1),(-1,-1)]#8个方向
for x in range(n):
for y in range(m):
if maps[x][y]==1:
dp[x][y]=9
continue
for dx,dy in direct:
new_x=x+dx
new_y=y+dy
if (new_x>=0 and new_y>=0 and new_x<n and new_y<m):#不能等于n和m因为列表从0开始取
dp[x][y]+=maps[new_x][new_y]#没越界则加上原地图上值,有地雷则加一,没地雷则加零
for x in range(n):
for y in range(m):
print(dp[x][y],end=' ')
print()
# for line in dp:
# print(' '.join(map(str,line)))
需要注意的是
1.这里用了*map直接构建列表。
2.判断是否越界的时候注意下标是从0开始,也就是等于n,m也算越界