[蓝桥杯2020初赛] 蛇形填数

题目描述

如下图所示，小明用从1开始的正整数“蛇形”填充无限大的矩阵。
1 2 6 7 15 …
3 5 8 14 …
4 9 13 …
10 12 …
11 …
…
容易看出矩阵第二行第二列中的数是 5。
请你计算矩阵中第 20 行第 20 列的数是多少？

输出格式

这是一道结果填空题，你只需要算出结果后提交即可。本题的结果为一个整数，在提交答案时只输出这个整数，输出多余的内容将无法得分。

#[蓝桥杯2020初赛] 蛇形填数
#1,5,13,25,41,61,85 发现规律
s=1
for i in range(20):
    s=s+4*i
print(s)

参考代码1

m, n, res = 0, 0, 0     
end = 20    
while True:     
    res += 1    
    if m == end-1 and n == end-1:
        break   
    if (m + n) & 1:   
        m += 1      
        if n > 0:
            n -= 1  
    else:
        n += 1      
        if m > 0:
            m -= 1  
print(res)