Python-CSP 201512-2 消除类游戏

一、题目

二、算法思想

• 使用二维列表来存储输入的棋子，进行消除的操作时，需要遍历所有的棋子。
• 对于第i+1行第j+1列的棋子来说，只需要从当前棋子开始往右遍历第i+1行，判断是否有连续3个及3个以上的同编号棋子；
• 从当前棋子开始往下遍历第j+1列，判断是否有连续3个及3个以上的同编号棋子；如果行满足时就记录对应行连续棋子的索引，如果列满足时就记录对应列连续棋子的索引。
• 不能边判断边进行消除的动作，否则将会出现行/列消除了，但列/行没有被消除。
• 例：当遍历到第1行第3列的棋子时，会先检测到第3行满足了条件，要是将第3列都置0，那么第2行就不再符合消除的条件了。

• 定义两个函数delete_hang和delete_lie，分别用于记录第i+1行第j+1列的棋子右侧和下方哪些棋子需要进行删除，并返回对应棋子在矩阵chart中的下标列表。
• delete_hang函数需要传入三个参数，分别为棋盘矩阵、棋子的行索引i、棋子的列索引j。
• 在自定义函数中i和j的值不进行变更，定义一个列表pos用于存储需要消除的棋子坐标，对相同行不同列的棋子进行遍历
• 列索引k遍历范围为[j，len(chart[i])-1]，len(chart[i])为矩阵chart的列数
• 如果此时的第k+1行j+1列棋子编号与传入的棋子编号相同时，就将此时的棋子行列索引添加到pos列表；
• 如果出现了第一次的不相同时，不符合连续棋子颜色相同的条件，可以直接退出遍历。
• 最后再判断已有的索引列表中，连续棋子的个数是否小于3，如果小于3时不符合消除的条件，如果大于等于3时就返回索引列表pos用于主函数的消除操作。

delete_lie函数与delete_hang的算法类似，不同的地方是行索引k遍历的范围为[i，len(chart)-1]，len(chart)为矩阵chart的行数，遍历时变化的位置为行索引。

• 在主函数中输入数据后，先对矩阵chart的每一个棋子进行遍历，通过函数delete_hang和delete_lie进行行消除和列消除的判断，并使用extend方法将函数返回列表的元素添加到position列表的末尾。
• 然后对索引列表position进行遍历，使chart矩阵需要消除的棋子位置置0，最后以空格为每行数据的间隔符逐行输出chart矩阵。

三、代码

# coding=utf-8
# 作者：小狐狸
# 题目：消除类游戏
def delete_hang(chart,i,j):
    '''找出当前棋子右侧需要消除的相同编号且连续的棋子索引'''
    pos = [] 
    for k in range(j,len(chart[i])): #遍历第i+1行j+1列的棋子及其右侧棋子
        if chart[i][j]==chart[i][k]:
            pos.append((i,k)) #i为行，k为列
        else: #不连续时
            break
    if len(pos)<3: #不满足消除条件时，不返回索引
        return []
    else:
        return pos
    
def delete_lie(chart,i,j):
    '''找出当前棋子下方需要消除的相同编号棋子索引'''
    pos = []
    for k in range(i,len(chart)): #遍历第i+1行j+1列的棋子及其下方棋子
        if chart[i][j]==chart[k][j]:
            pos.append((k,j))
        else: #不连续时
            break
    if len(pos)<3: #不满足消除条件时，不返回索引
        return []
    else:
        return pos
    
if __name__=='__main__':
    n,m = input().split()
    n = int(n)
    m = int(m)
    chart = []  #棋盘矩阵
    for i in range(n):
        lis = list(map(int,input().split()))  #当前行
        chart.append(lis)
##    print(chart)
    position = []   #需要被删除的棋子在棋盘矩阵的索引列表
    for i in range(n):
        for j in range(m):
            position.extend(delete_hang(chart,i,j)) #行消除
            position.extend(delete_lie(chart,i,j))  #列消除
##    print(position)
    for pos in position: #消除被标记的棋子
        i = pos[0]
        j = pos[1]
        chart[i][j] = 0     
    for i in chart: #输出
        print(' '.join(str(j) for j in i))