CCF 201412-2 Z字形扫描（Python）

CCF 201412-2 Z字形扫描（Python）

一、问题描述

二、满分代码

n = int(input())
s = []

for i in range(n):
    a = list(map(int,input().split()))
    s += [a]

x,y = 0,0
d = [0 for _ in range(n*n)]
#设置三个判断下一步动作的布尔
right=True
l_down=False
r_up=False
for i in range(n*n):
    
    d[i]=s[x][y]

    if right==True:
        if y!=n-1 and x!=n-1:
            y+=1
            right=False
            l_down=True
        elif y==n-1 and x==n-1:
            right=False

            l_down=False
            r_up=False
        else:
            x -= 1
            right=False
            l_down=True
        
    elif l_down==True:
        if y!=0 and x!=n-1:
            x += 1
            y -= 1
            l_down=True
        elif x == n-1:
            y += 1
            if y == n-1:
                right=False
                l_down=False
                r_up=False
            else:
                l_down=False
                r_up=True
            
        else:
            x += 1
            r_up = True
            l_down= False
        
        
    elif r_up==True:
        if x!= 0 and y!=n-1:
            x -= 1
            y += 1
            r_up=True
        elif x == 0 and y!=n-1:
            y+=1
            r_up=False
            l_down = True
        else:
            x += 1
            r_up = False
            l_down = True

for i in range(n*n):
    print(d[i],end=' ')

三、结果