蓝桥杯 蛇形填数

题目描述

本题为填空题，只需要算出结果后，在代码中使用输出语句将所填结果输出即可。

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

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

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

运行限制

• 最大运行时间：1s
• 最大运行内存: 128M

题解

 解答

import os
import sys

# 请在此输入您的代码
res = 0
be = 1
for i in range(1,20):
  res = be+i*4
  be = res
print(res)

或者 

dp = []
dp.append(1)
for i in range(1,21):
    dp.append(dp[i-1]+(i-1)*4)
print(dp[-1])