题解 | #蛇形矩阵#
蛇形矩阵
http://www.nowcoder.com/practice/649b210ef44446e3b1cd1be6fa4cab5e
import sys
def run(a):
b = [[] for _ in range(a)]
c = list(range(1, sum(list(range(1,a+1)))+1))
for i in range(1, a+1):
d, c = c[:i], c[i:]
for j in range(len(d)):
b[len(d)-j-1].append(str(d[j]))
b = [" ".join(_) for _ in b]
b = "\n".join(b)
return b
for line in sys.stdin:
_a = line.strip()
print(run(int(_a)))