CCF Python题解(100分)201412-2 Z字形扫描

n = int(input())
nums = []
for i in range(n):
nums.append(list(map(int, input().split())))
i = 0
j = 0
print(nums[0][0], end=" ")
flag = "right"
while True:
if flag == 'right':
j += 1
if n == 1:
break
if i == 0:
flag = "left-down"
if i == n - 1:
flag = "right-up"
if i == n - 1 and j == n - 1:
print(nums[i][j])
break
print(nums[i][j], end=" ")

if flag == 'down':
i += 1
if j == 0:
flag = "right-up"
if j == n - 1:
flag = "left-down"
print(nums[i][j], end=" ")

if flag == "left-down":
i += 1
j -= 1
if j == 0:
flag = "down"
if i == n - 1:
flag = "right"
print(nums[i][j], end=" ")

if flag == "right-up":
i -= 1
j += 1
if i == 0:
flag = "right"
if j == n - 1:
flag = "down"

print(nums[i][j], end=" ")