Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- def countMine(mat, n):
- rows = n
- columns = n
- #lista = ['#' if elem=='#' else 0 for elem in [item for item in mat ]]
- bombs=[]
- for i in range(len(mat)):
- for j in range(len(mat[i])):
- if mat[i][j]=='#':
- bombs.append((i,j));
- filed = [[0 for i in range(columns)] for j in range(rows)]
- for bomb_location in bombs:
- (bomb_row, bomb_col) = bomb_location
- filed[bomb_row][bomb_col] = '#'
- row_range = range(bomb_row - 1, bomb_row + 2)
- col_range = range(bomb_col - 1, bomb_col + 2)
- for i in row_range:
- for j in col_range:
- if (0<=i<rows and 0<=j<columns and filed[i][j] != '#'):
- filed[i][j] +=1
- return filed
- def main():
- n=int(input())
- matrica = []
- for i in range(n):
- matrica.append([])
- list = input().split(" ")
- for j in range(n):
- matrica[i].append(list[j])
- nova_mat = countMine(matrica, n)
- for i in range(len(nova_mat)):
- print(" ".join(map(str,nova_mat[i])))
- #print(nova_mat[i])
- if __name__ == "__main__":
- main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement