CCF 201512-2 消除类游戏 python 满分

题目叙述

问题描述：略

输入格式：略

输出格式：略

样例

样例输入
4 5
2 2 3 1 2
3 4 5 1 4
2 3 2 1 3
2 2 2 4 4
样例输出
2 2 3 0 2
3 4 5 0 4
2 3 2 0 3
0 0 0 4 4
样例说明
　　棋盘中第4列的1和第4行的2可以被消除，其他的方格中的棋子均保留。
样例输入
4 5
2 2 3 1 2
3 1 1 1 1
2 3 2 1 3
2 2 3 3 3
样例输出
2 2 3 0 2
3 0 0 0 0
2 3 2 0 3
2 2 0 0 0
样例说明
　　棋盘中所有的1以及最后一行的3可以被同时消除，其他的方格中的棋子均保留。

满分证明

解题思路

建立标志，十字形查找，逐个遍历。

满分代码

n, m = map(int, input().split())
npq = [[0] * (m + 2) for _ in range(n + 2)]
for i in range(n):
    npq[i + 1:i + 2][0][1:- 1] = list(map(int, input().split()))
tnpq = [[1] * (m + 2) for _ in range(n + 2)]

for i in range(1, n + 1):
    for j in range(1, m + 1):
        if (npq[i:i + 1][0][j - 1] == npq[i:i + 1][0][j]) & (npq[i:i + 1][0][j + 1] == npq[i:i + 1][0][j]):
            tnpq[i:i + 1][0][j - 1] = 0
            tnpq[i:i + 1][0][j] = 0
            tnpq[i:i + 1][0][j + 1] = 0
        if (npq[i - 1:i][0][j] == npq[i:i + 1][0][j]) & (npq[i + 1:i + 2][0][j] == npq[i:i + 1][0][j]):
            tnpq[i - 1:i][0][j] = 0
            tnpq[i:i + 1][0][j] = 0
            tnpq[i + 1:i + 2][0][j] = 0

for i in range(1, n + 1):
    for j in range(1, m + 1):
        if tnpq[i:i + 1][0][j]:
            print(npq[i:i + 1][0][j], end=" ")
        else:
            print(0, end=" ")
    print()

优化代码

      将  if (npq[i:i + 1][0][j - 1] == npq[i:i + 1][0][j]) & (npq[i:i + 1][0][j + 1] == npq[i:i + 1][0][j]):
       改为  if npq[i:i + 1][0][j - 1] == npq[i:i + 1][0][j] ==npq[i:i + 1][0][j + 1]:

需者自取传送门(∩ᄑ_ᄑ)⊃━☆【CCF 2013-2021】本博主整理历年至少前两题 python 满分代码目录