蛇形填数594

题目

如下图所示，小明用从 1开始的正整数“蛇形”填充无限大的矩阵。

1 2 6 7 15 ...
3 5 8 14 ...
4 9 13 ...
10 12 ...
11 ...
...

容易看出矩阵第二行第二列中的数是 5。请你计算矩阵中第 20 行第 20 列的数是多少？

代码

import os
import sys

# 请在此输入您的代码
#1   2  6  7 15 16 28
#3   5  8 14 17 27
#4   9 13 18 26
#10 12 19 25
#11 20 24
#21 23
#22
#题目要求20行20列的数，是对角线上的数。可以看到1,5,13,25.
#1=1*1，5=2*2+1*1,13=3*3+2*2,25=4*4+3*3
#即对角线上的数=当前行^2+(当前行-1)^2
a = 20*20+19*19
print(a)