蓝桥杯 蛇形填数

问题描述

如下图所示，小明用从 1 开始的正整数“蛇形”填充无限大的矩阵。
1 2 6 7 15 …
3 5 8 14 …
4 9 13 …
10 12 …
11 …
…

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

【答案提交】

这是一道结果填空的题，你只需要算出结果后提交即可。

本题的结果为一个整数，在提交答案时只填写这个整数，填写多余的内容将无法得分

思路分析及代码实现

这道题是一道填空题，然后我们可以通过观察找规律来得出答案
通过观察发现：
第一行第一列 ： 1 1^2 + (1-1)^2 = 1
第二行第二列： 5 2^2 + (2-1)^2 = 5
第三行第三列： 13 3^2 + (3-1)^2 = 13
第四行第四列： 25 4^2 + (4-1)^2 = 25
第五行第五列： 41 5^2 + (5-1)^2 = 41

所以通过找规律发现：第n行第n列 = n^2 + (n-1)^2
第二十行第二十列 ： 20*2 + (20-1)^2 = 761
答案：761

当然也可以用代码实现，就是当到达边界时，判断它应该向右走还是向下走，向右走完就直接向左下走，向下走完就直接向右上走，因为没有边界，所以在走的过程中只要x,y = 19,19（从0，0开始的）就直接停下输出就可以了

dir = 0  # 0代表向右，1代表向下
a = [[0 for i in range(100)] for j in range(100)]
x, y = 0, 0
num = 1
while True:
    if x == 19 and y == 19:
        break
    while dir == 0:
        x, y = x, y+1
        num += 1
        while y != 0:
            x += 1
            y -= 1
            num += 1
            if x == 19 and y == 19:
                print(num)
                break
        dir = 1
    while dir == 1:
        x, y = x+1, y
        num += 1
        while x != 0:
            x -= 1
            y += 1
            num += 1
            if x == 19 and y == 19:
                print(num)
                break
        dir = 0