2015-12 消除类游戏 python 100分

问题描述
　　消除类游戏是深受大众欢迎的一种游戏，游戏在一个包含有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。

import copy
n,m = map(int,input().split()) # 行数和列数

# 消除数组
list_all = []
for i in range(n):
        list_1 = list(map(int, input().split()))
        list_all.append(list_1)


list_all_1 = copy.deepcopy(list_all) #进行行消除的数组
list_all_2 = copy.deepcopy(list_all) #进行列消除的数组

#定义行的消除方式
def delete_hang(l,m):
    num = 1
    for i in range(m-1):
        if l[i] == l[i+1]:#如果前一个数等于后一个数
            num = num + 1
            if i==m-2:
                if num >= 3:  # 如果连续3个以上的数相等
                    l[i+2-num:m] = [0] * num
        else:
            if num >= 3:#如果连续3个以上的数相等
                l[i+1-num:i+1] = [0]*num
            num = 1
            continue
    return l

#定义列的消除方式 和行类似
def delete_lie(list_all,n,m):
    for j in range(m):
        num = 1
        for i in range(n-1):
            if list_all[i][j] == list_all[i+1][j]:
                num = num+1
                if i==n-2:
                    if num >= 3:
                        # 消除为0
                        for k in range(i+2-num,n):
                            list_all[k][j]=0
            else:
                if num >= 3:
                    for k in range(i + 1 - num, i + 1):
                        list_all[k][j] = 0
                num = 1
                continue
    return list_all

#按照行消除-->list_all_1
for i in list_all_1:
    delete_hang(i,m)
#按照列消除-->list_all_2
delete_lie(list_all_2,n,m)

# 在原数组上进行0替换
for i in range(n):
    for j in range(m):
        if list_all_1[i][j] == 0:
            list_all[i][j] = list_all_1[i][j]
        elif list_all_2[i][j] == 0:
            list_all[i][j] = list_all_2[i][j]
        else:
            continue

for i in range(n):
    for j in range(m):
        print(list_all[i][j], end=' ')
    if i != n-1:
        print()