CCF CSP 201412-1/2 门禁系统/Z字形扫描 python

201412-1 门禁系统

n=int(input())
data = list(map(int,input().split()))
reader={}
for i in range(n):
    if data[i] not in reader:
        reader[data[i]] = 0
    reader[data[i]] += 1
    print(reader[data[i]],end=" ") 

201412-2 Z字形扫描

n=int(input())
data = []
for i in range(n):
    data.append([])
    data[i]=list(map(int,input().split()))

i,j = 0,0
result = []
up = True

for t in range(n*n):
    result.append(data[i][j])
    if up:
        if j == n-1:
            i+=1
            up = False
        elif i == 0:
            #init up
            j+=1
            up = False
        else:
            i-=1
            j+=1
    else:
        if i == n-1:
            j+=1
            up = True
        elif j == 0:
            i+=1
            up = True
        else:
            i+=1
            j-=1
for i in range(len(result)):
    print(result[i],end=" ")