CCF Python题解(100分)201512-2 消除类游戏

import copy

n, m = map(int, input().split())
data = []
for i in range(n):
data.append(input().split())
newdata = copy.deepcopy(data)
for row in range(n):
for col in range(m - 2):
if data[row][col + 1] == data[row][col] and data[row][col + 2] == data[row][col]:
newdata[row][col + 1] = '0'
newdata[row][col] = '0'
newdata[row][col + 2] = '0'
for col1 in range(col + 3, m):
if data[row][col1] == data[row][col]:
newdata[row][col1] = '0'
else:
break

for col in range(m):
for row in range(n - 2):
if data[row][col] == data[row + 1][col] and data[row + 2][col] == data[row][col]:
newdata[row][col] = '0'
newdata[row + 1][col] = '0'
newdata[row + 2][col] = '0'
for row1 in range(row + 3, n):
if data[row1][col] == data[row][col]:
newdata[row1][col] = '0'
else:
break

for i in range(n):
for j in range(m):
print(newdata[i][j], end=" ")
print()