问题描述
试题编号: | 201412-2 |
试题名称: | Z字形扫描 |
时间限制: | 2.0s |
内存限制: | 256.0MB |
问题描述: | 问题描述 在图像编码的算法中,需要将一个给定的方形矩阵进行Z字形扫描(Zigzag Scan)。给定一个n×n的矩阵,Z字形扫描的过程如下图所示: 输入格式 输入的第一行包含一个整数n,表示矩阵的大小。 输出格式 输出一行,包含n×n个整数,由空格分隔,表示输入的矩阵经过Z字形扫描后的结果。 样例输入 4 样例输出 1 5 3 9 7 3 9 5 4 7 3 6 6 4 1 3 评测用例规模与约定 1≤n≤500,矩阵元素为不超过1000的正整数。 |
代码
#Z字形扫描
n = int(input())
rect = []
for i in range(n):
rect.append(list(map(int,input().split())))
i,j = 0,0
line = []#经历过的数字
up = True#方向向上
for t in range(n*n):
line.append(rect[j][i])
#碰壁后就转向
if up:#右上
if i==n-1: # 碰到了右边界
j += 1 # 向下走一格
up=False # 转向
elif j==0: # 碰到了上边界
i += 1 # 向右走一格
up=False # 转向
else: # 没碰到边界
i += 1 # 向右上斜着走一格
j -= 1
else:#左下
if j==n-1: # 碰到了下边界
i += 1 # 向右走一格
up=True # 转向
elif i==0: # 碰到了左边界
j += 1 # 向下走一格
up=True # 转向
else: # 没碰到边界
i -= 1 # 向左下斜着走一格
j += 1
print(" ".join(map(str,line)))