Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def read_matrix(s):
- m = []
- for r in range(s):
- txt = input()
- txt_list = []
- for char in txt:
- txt_list.append(char)
- m.append(txt_list)
- return m
- def valid_move(move):
- r = move[0]
- c = move[1]
- if r in range(len(matrix)) and c in range(len(matrix)):
- return True
- return False
- def remove_k(m, s):
- counter = 0
- for r in range(s):
- for c in range(s):
- current_cell = m[r][c]
- if current_cell == "K":
- move_1 = (r + 2, c + 1)
- move_2 = (r + 2, c - 1)
- move_3 = (r + 1, c + 2)
- move_4 = (r - 1, c + 2)
- move_5 = (r - 2, c + 1)
- move_6 = (r - 2, c - 1)
- move_7 = (r + 1, c - 2)
- move_8 = (r - 1, c - 2)
- moves = [move_1, move_2, move_3, move_4, move_5, move_6, move_7, move_8]
- for move in moves:
- if valid_move(move):
- target_cell = m[move[0]][move[1]]
- if target_cell == current_cell:
- m[move[0]][move[1]] = "0"
- counter += 1
- #for r in m:
- # print(f'{"".join(x for x in r)}')
- return counter
- size = int(input())
- matrix = read_matrix(size)
- print(remove_k(matrix, size))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement