CSP201604-2 俄罗斯方块 (Python)

题目

试题编号：	201604-2
试题名称：	俄罗斯方块
时间限制：	1.0s
内存限制：	256.0MB

问题描述
　　俄罗斯方块是俄罗斯人阿列克谢·帕基特诺夫发明的一款休闲游戏。
　　游戏在一个15行10列的方格图上进行，方格图上的每一个格子可能已经放置了方块，或者没有放置方块。每一轮，都会有一个新的由4个小方块组成的板块从方格图的上方落下，玩家可以操作板块左右移动放到合适的位置，当板块中某一个方块的下边缘与方格图上的方块上边缘重合或者达到下边界时，板块不再移动，如果此时方格图的某一行全放满了方块，则该行被消除并得分。
　　在这个问题中，你需要写一个程序来模拟板块下落，你不需要处理玩家的操作，也不需要处理消行和得分。
　　具体的，给定一个初始的方格图，以及一个板块的形状和它下落的初始位置，你要给出最终的方格图。
　　
输入格式
　　输入的前15行包含初始的方格图，每行包含10个数字，相邻的数字用空格分隔。如果一个数字是0，表示对应的方格中没有方块，如果数字是1，则表示初始的时候有方块。输入保证前4行中的数字都是0。
　　输入的第16至第19行包含新加入的板块的形状，每行包含4个数字，组成了板块图案，同样0表示没方块，1表示有方块。输入保证板块的图案中正好包含4个方块，且4个方块是连在一起的（准确的说，4个方块是四连通的，即给定的板块是俄罗斯方块的标准板块）。
　　第20行包含一个1到7之间的整数，表示板块图案最左边开始的时候是在方格图的哪一列中。注意，这里的板块图案指的是16至19行所输入的板块图案，如果板块图案的最左边一列全是0，则它的左边和实际所表示的板块的左边是不一致的（见样例）
　　
输出格式
　　输出15行，每行10个数字，相邻的数字之间用一个空格分隔，表示板块下落后的方格图。注意，你不需要处理最终的消行。
　　
样例输入
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 1 0 0
　　0 0 0 0 0 0 1 0 0 0
　　0 0 0 0 0 0 1 0 0 0
　　1 1 1 0 0 0 1 1 1 1
　　0 0 0 0 1 0 0 0 0 0
　　0 0 0 0
　　0 1 1 1
　　0 0 0 1
　　0 0 0 0
　　3
　　
样例输出
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 0 0 0
　　0 0 0 0 0 0 0 1 0 0
　　0 0 0 0 0 0 1 0 0 0
　　0 0 0 0 0 0 1 0 0 0
　　1 1 1 1 1 1 1 1 1 1
　　0 0 0 0 1 1 0 0 0 0

程序代码

# 输入初始方格图(先添加4行空行和底端边界)
numbers = []
for i in range(4):
    numbers.append([0 for i in range(10)])
for i in range(15):
    temp = list(map(int,input().split()))
    numbers.append(temp)
numbers.append([1 for i in range(10)])

# 输入新板块形状
new = []
for i in range(4):
    temp = list(map(int,input().split()))
    new.append(temp)

# 输入下落位置
n = int(input())

# 确定新板块方格位置并加入顶端四行空行中
check = []
check_0 = []
for i in range(4):
    for j in range(4):
        if new[i][j] == 1:
            check.append([i,j+n-1])
            # 寻找新板块中下方为空的方格用于判断下移距离
            if i == 3 or new[i+1][j] == 0:
                check_0.append([i,j+n-1])

# 判断合法的下移行数
down = 0
flag = True # 循环标记
while flag:
    for x_y in check_0:
        x,y = x_y
        if numbers[x+1][y] != 0:
            flag = False
    down += 1
    for i in range(len(check_0)):
        check_0[i][0] += 1

# 改变下落最终位置的数字
for i in range(4):
    x,y = check[i]
    numbers[x + down-1][y] = 1

# 输出
for i in range(4,19):
    for j in range(10):
        print(numbers[i][j],end=" ")
    print()