时间限制: | 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的正整数。 |
逻辑不是很完美。。。可以用dfs写的更好看些。简单的一个规律,从当前位置按照右上,右,左下,下的方向走,往右上,左下的时候碰壁则换方向,往右和往下则走一步换方向。还有,起点和终点固定。。
def comp(t):
if t<0 or t>=n:
return False
return True
while True:
try:
dir = [[-1,1],[0,1],[1,-1],[1,0]]
vis = [[0 for i in range(505)]for j in range(505)]
n = int(input())
m = []
for i in range(n):
#for j in range(n):
m.append([int(v) for v in input().split()])
#print(dir[1][0])
cnt = 0
x=y=0
while cnt<(n**2)-1:
for i in range(4):
while cnt<(n**2):
if comp(x+dir[i][0]) and comp(y+dir[i][1]):
if not vis[x+dir[i][0]][y+dir[i][1]]:
print(m[x][y],end=' ')
x+=dir[i][0]
y+=dir[i][1]
vis[x][y] = 1
cnt+=1
if i==1 or i==3:
break
else:
break
else:
break
print(m[n-1][n-1])
except:
break