蓝桥杯 2020蛇形填数 python

题目描述

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

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

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

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

mp=[[0 for i in range(50)]for j in range(50)]
x=1
y=1
tot=0
while x+y<=40:
     tot+=1
     mp[x][y]=tot
     if x==1:
          y+=1
          tot+=1
          mp[x][y]=tot
          dx=1
          dy=-1
     if y==1:
          x+=1
          tot+=1
          mp[x][y]=tot
          dx=-1
          dy=1
     x+=dx
     y+=dy
print(mp[20][20])