本文已参与「新人创作礼」活动，一起开启掘金创作之路

蛇形填数

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

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

lst = [[] for i in range(40)]
num = 1
lst[0].append(1)
for i in range(2,40):    # 控制行数
    if i%2==0:   # 当行数是偶数，从右上到左下 
        for j in range(i):  
            num += 1
            lst[j].append(num)
    else:       # 当行数是奇数 从左下到右上
        for j in range(i):
            num += 1
            lst[i-j-1].append(num)
print(lst[19][19])

# for i in range(len(lst)):
#    print(i+1,lst[i])

寻找2020

【问题描述】 小蓝有一个数字矩阵，里面只包含数字 0 和 2。小蓝很喜欢 2020，他想找 到这个数字矩阵中有多少个 2020 。 小蓝只关注三种构成 2020 的方式： • 同一行里面连续四个字符从左到右构成 2020。 • 同一列里面连续四个字符从上到下构成 2020。 • 在一条从左上到右下的斜线上连续四个字符，从左上到右下构成 2020。 例如，对于下面的矩阵： 220000 000000 002202 000000 000022 002020 一共有 5 个 2020。其中 1 个是在同一行里的，1 个是在同一列里的，3 个 是斜线上的。 小蓝的矩阵比上面的矩阵要大，由于太大了，他只好将这个矩阵放在了一个文件里面，在试题目录下有一个文件 2020.txt，里面给出了小蓝的矩阵。 请帮助小蓝确定在他的矩阵中有多少个 2020。

nums = [[2, 2, 0, 0, 0, 0],
        [0, 0, 0, 0, 0, 0],
        [0, 0, 2, 2, 0, 2],
        [0, 0, 0, 0, 0, 0],
        [0, 0, 0, 0, 2, 2],
        [0, 0, 2, 0, 2, 0],
        ]

n = 0
for i in range(len(nums)):
    for j in range(len(nums[0])):
        if j+4 <= len(nums[0]):   # 行中2020 
            a = nums[i][j:j+4]
            if a == [2,0,2,0]:
                n+=1
        if i+3<len(nums):  # 列中出现2020
            b = nums[i][j]*1000+nums[i+1][j]*100+nums[i+2][j]*10+nums[i+3][j]
            if b == 2020: 
                n+=1
        if i+3<len(nums) and j+3 < len(nums[0]):   # 从左上到右下出现2020
            c = nums[i][j]*1000+nums[i+1][j+1]*100+nums[i+2][j+2]*10+nums[i+3][j+3]
            if c == 2020:
                n+=1
print(n)