Python入门习题（34）——CCF CSP认证考试真题：消除类游戏

问题描述

试题编号： 201512-2
试题名称： 消除类游戏
时间限制： 1.0s
内存限制： 256.0MB

问题描述
　　消除类游戏是深受大众欢迎的一种游戏，游戏在一个包含有n行m列的游戏棋盘上进行，棋盘的每一行每一列的方格上放着一个有颜色的棋子，当一行或一列上有连续三个或更多的相同颜色的棋子时，这些棋子都被消除。当有多处可以被消除时，这些地方的棋子将同时被消除。
　　现在给你一个n行m列的棋盘，棋盘中的每一个方格上有一个棋子，请给出经过一次消除后的棋盘。
　　请注意：一个棋子可能在某一行和某一列同时被消除。
输入格式
　　输入的第一行包含两个整数n, m，用空格分隔，分别表示棋盘的行数和列数。
　　接下来n行，每行m个整数，用空格分隔，分别表示每一个方格中的棋子的颜色。颜色使用1至9编号。
输出格式
　　输出n行，每行m个整数，相邻的整数之间使用一个空格分隔，表示经过一次消除后的棋盘。如果一个方格中的棋子被消除，则对应的方格输出0，否则输出棋子的颜色编号。
样例输入
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可以被同时消除，其他的方格中的棋子均保留。
评测用例规模与约定
　　所有的评测用例满足：1 ≤ n, m ≤ 30。

解题思路

1. 逐列地考察，看该列上有没有要消除的棋子。
2. 逐行地考察，看该行上有没有要消除的棋子。
3. 考察期间，不能立即执行消除动作，而是用一个n行m列矩阵erased来记录哪些位置的棋子要消除。
4. 考察完毕后，据矩阵erased来消除棋子。
5. 如何判断一行上有没有要消除的棋子？做法是，从第2列到第m-1列，如果该列棋子的颜色与其左右两边棋子的颜色相同，那么这3个棋子要消除。这么做，连续4个，5个同色棋子会不会一起消除？答案是，会。连续4个是两组连续3个组成的，这两组连续3个有重合。同样做法可以应用到判断一列上要消除的棋子。

参考答案

n, m = [int(s) for s in input().split()]
board = []
for i in range(n):
    board.append(input().strip().split())
# print(board)

erased_line = [False] * m
erased = []  #哪些位置的棋子要消除？要消除的棋子所在位置，erased元素值设为True
for i in range(n):
    erased.append(erased_line[:])
#考察每一列
for column in range(m):
    for row in range(1, n-1):
        if board[row - 1][column] == board[row][column] and board[row][column] == board[row + 1][column]:
            erased[row - 1][column] = True
            erased[row][column] = True
            erased[row + 1][column] = True
#考察每一行
for row in range(n):
    for column in range(1, m-1):
        if board[row][column - 1] == board[row][column] and board[row][column] == board[row][column + 1]:
            erased[row][column - 1] = True
            erased[row][column] = True
            erased[row][column + 1] = True
#消除棋子
for row in range(n):
    for column in range(m):
        if erased[row][column]:
            board[row][column] = '0'

for row in range(n):
    print(' '.join(board[row]))

测试用例

1. 题目描述中给出的第一组测试用例，覆盖了一行上有3个棋子要消除的情形，也覆盖了一列上有3个棋子要消除的情形。

2. 题目描述中给出的第二组测试用例，覆盖了一行上有4个（多于3个）棋子要消除的情形，也覆盖了一列上要消除的棋子和一行上要消除的棋子交叉的情形。

3. n=1或m=1的边界情形。
   样例输入
   1 1
   5
   样例输出
   5

4. n=2或m=2的边界情形。
   样例输入
   2 2
   3 3
   3 3
   样例输出
   3 3
   3 3
   这一组测试用例也表明，上下左右连成4个同色棋子是不会被消除的。

5. 全部棋子都消除。
   样例输入
   2 3
   1 1 1
   1 1 1
   样例输出
   0 0 0
   0 0 0

6. 全部棋子不消除。
   样例输入
   3 2
   3 3
   3 3
   4 4
   样例输出
   3 3
   3 3
   4 4

小结

1. 人观察图像的能力特别强，不见得是逐行逐列地来搜索要消除的棋子。计算机程序的做法总是把大任务分解成小任务。这里，把整幅图像的消除任务分解成逐行的消除任务和逐列的消除任务。
2. 要注意：考察期间，不能立即执行消除动作。这样会遗漏一些可以消除的棋子。不相信的话，你可以拿题目描述给出的第2组测试用例来试验下。