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=" ")