蓝桥杯——蛇形填数

使用python求解蛇形填数

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

解： 将后面的数依次填充出来找规律，由于问题只问的n行n列，所以我们只关注这样一个位置（n行n列）
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
n行n列的数依次是： 1 5 13 25 41 相邻两者相差： 4 8 12 16

from functools import reduce
def add(x,y):
    return x+y
print(reduce(add,[4*i for i in range(1,20)])+1)	#+1因为第一个数为1
结果：761

注意：reduce把一个函数作用在一个序列[x1, x2, x3, …]上，这个函数必须接收两个参数，reduce把结果继续和序列的下一个元素做累积计算