CCF CSP认证201412-2 Z字扫描

# n,m=input().split()
# n,m=int(n),int(m)
import time
# c1 = time.perf_counter()
n = int(input())
data = list()
for i in range(n):
    s=input().split()
    for j in range(n):
        s[j]=int(s[j])
    data.append(s)
result = list()
i = 0#hang
j = 0
flag = 1
c1 = time.perf_counter()
while(len(result)<n*n):
    if(j==0 and flag==1):
        # if(i%2==0):
            if(i+1<n):
                result.append(data[j][i])
                i += 1
                flag = -1
                continue
            else:
                result.append(data[j][i])
                j += 1
                flag = -1
                continue
    elif(i==0 and flag==-1):
            if (j + 1 < n):
                result.append(data[j][i])
                j += 1
                flag = 1
                continue
            else:
                result.append(data[j][i])
                i += 1
                flag = 1
                continue
    elif (i == n-1 and j!=0 and flag == 1):
            if (j + 1 < n):
                result.append(data[j][i])
                j += 1
                flag = -1
                continue
            else:
                result.append(data[j][i])
                i += 1
                flag = -1
                continue
    elif (j == n-1 and i!=0 and flag == -1):
            if (i + 1 < n):
                result.append(data[j][i])
                i += 1
                flag = 1
                continue
            else:
                result.append(data[j][i])
                j += 1
                flag = 1
                continue
    elif(flag==-1):
        result.append(data[j][i])
        i -= 1
        j += 1
    else:
        result.append(data[j][i])
        i += 1
        j -= 1
print(result)
print(time.perf_counter()-c1)