CSP201412-2 Z字形扫描 (Python)

题目

试题编号：	201412-2
试题名称：	Z字形扫描
时间限制：	2.0s
内存限制：	256.0MB

问题描述
　　在图像编码的算法中，需要将一个给定的方形矩阵进行Z字形扫描(Zigzag Scan)。给定一个n×n的矩阵，Z字形扫描的过程如下图所示：

　　对于下面的4×4的矩阵，
　　1 5 3 9
　　3 7 5 6
　　9 4 6 4
　　7 3 1 3
　　对其进行Z字形扫描后得到长度为16的序列：
　　1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3
　　请实现一个Z字形扫描的程序，给定一个n×n的矩阵，输出对这个矩阵进行Z字形扫描的结果。
　　
输入格式
　　输入的第一行包含一个整数n，表示矩阵的大小。
　　输入的第二行到第n+1行每行包含n个正整数，由空格分隔，表示给定的矩阵。
　　
输出格式
　　输出一行，包含n×n个整数，由空格分隔，表示输入的矩阵经过Z字形扫描后的结果。
　　
样例输入
　　4
　　1 5 3 9
　　3 7 5 6
　　9 4 6 4
　　7 3 1 3
　　
样例输出
　　1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3
　　
评测用例规模与约定
　　1≤n≤500，矩阵元素为不超过1000的正整数。

程序代码

# 输入矩阵大小
n = int(input())

# 矩阵
numbers = []

# 输出顺序
output = []

# 逐行输入矩阵
for i in range(n):
    # 输入行并格式转换
    temp = list(map(int,input().split()))
    numbers.append(temp)

# 初始化当前位置
x,y = 0,0

# 初始化扫描方向标记
up,down = True,False

# 当未到达末尾元素坐标时
while [x,y] != [n-1,n-1]:
    #print(x,y)
    output.append(numbers[x][y])

    # 当此时在第一行
    if x == 0:
        if up == True: # 向上扫描到达第一行时
            # 当到达最后一列
            if y == n-1:
                x += 1
                up = False
                down =True
            else:
                y += 1
                up = False
        else: # 已经到达第一行且向右扫描后
            x += 1
            y -= 1
            down = True

    # 当此时在第一列
    elif y == 0:
        if down == True: # 向下扫描到达第一行时
        # 当到达最后一行
            if x == n-1:
                y += 1
                down = False
                up =True
            else:
                x += 1
                down = False
        else: # 已经到达第一列且向下扫描后
            x -= 1
            y += 1
            up = True

    # 当此时不在第一行也不在第一列时
    else:
        # 当向上扫描时
        if up == True:
            # 当到达最后一列
            if y == n-1:
                x += 1
                up = False
                down =True
            else:
                x -= 1
                y += 1
        else:  # 当向上扫描时(down = True)
            # 当到达最后一行
            if x == n-1:
                y += 1
                down = False
                up =True
            else:
                x += 1
                y -= 1

# 加入最后一个元素
output.append(numbers[n-1][n-1])

# 输出
for i in range(n*n):
    print(output[i],end=" ")