CCF201412-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的正整数。

 

问题链接：CCF201412-2 Z字形扫描

问题分析

明确四种走法：右、左下、下、右上

右和下对应两种接下来的走法，左下和右上对应三种接下来的走法，接下来只需要设置一定的条件，即可以让程序自行按照写好的规律扫描元素。请读者自行寻找一下题目中的这几个规律。我总结的规律保存在了下面代码中的path变量中。

用这种思路写的代码在测评上用了900多ms，快超时了，但是不用担心。考试时用python的话，限制的时常都是平时练习的10倍。

具体实现步骤请参考下面代码

满分代码

#输入数据，将表格存储到二维数组里面
n=int(input())
z=[]
for i in range(n):
    s=input().split()
    z+=[s]
#先将左上角的元素添加到输出序列中
out=[z[0][0]]
count=1
#定义扫描方法：
#前两个参数为该操作需要行移动多少，列移动多少。第三个为第一选择方案，最后面为备选方案。
path={'r':[0,1,'l_d','r_u'],
      'l_d':[1,-1,'d','r'],
      'd':[1,0,'r_u','l_d'],
      'r_u':[-1,1,'r','d']}
#初始操作为向右,所以假定第一个操作为r
i=0
j=0
be='r'
#结束条件是读入了n*n个数字,即全部扫描了一遍
while(count!=n*n): 
    i+=path[be][0]
    j+=path[be][1]
    out+=[z[i][j]]
    count+=1
    #下面两种情况是指向左下或者右上移动时如果不在边界位置。则继续按照这个方向移动
    if be=='r_u':
        if i-1>=0 and j+1<=n-1:
            continue
    if be=='l_d':
        if i+1<=n-1 and j-1>=0:
            continue
    #判断下一步如果采用第一种走法是否会出现越界情况。即行或者列大于n-1或者小于0都算越界。
    if (i+path[path[be][2]][0]<0)or(i+path[path[be][2]][0]>n-1)or(j+path[path[be][2]][1]<0)or(j+path[path[be][2]][1]>n-1):
        #越界即采用另一种走法
        be=path[be][3]
    else:
        #数组不越界即采用第一种走法
        be=path[be][2]
#输出结果
print(out[0],end='')
for i in range(1,len(out)):
    print('',out[i],end='')