CCFcsp-python-201412-2-Z字形扫描

思路：
1.在同一斜线上的下标和相同，建立字典存放下标和，以及对应的下标
2.下标和为偶数时，顺序为由大到小，进行排序
3.读出对应下标的值

n = int(input())
res = []
for i in range(n):
    inp = list(map(int, input().split()))
    res.append(inp)
dic = {}
# 把对应的下标 放到字典中
for i in range(n):
    for j in range(n):
        if i+j not in dic:
            dic[i+j] = [(i,j)]
        else:
            dic[i + j] .append((i, j))

# Z字形，为偶数时的下标对应由大到小，进行排序
for i in range(len(dic)):
    if i%2==0:
        temp = sorted(dic[i], key = lambda x:x[0],reverse=True)
        dic[i] = temp
# 读取对应的数
for i in range(len(dic)):
    for j in range(len(dic[i])):
        row = dic[i][j][0]
        col = dic[i][j][1]
        print(res[row][col],end=" ")