寻找2020

问题

小蓝有一个数字矩阵，里面只包含数字 0 和 2。小蓝很喜欢 2020，他想找到这个数字矩阵中有多少个 2020 。

小蓝只关注三种构成 2020 的方式：
• 同一行里面连续四个字符从左到右构成 2020。
• 同一列里面连续四个字符从上到下构成 2020。
• 在一条从左上到右下的斜线上连续四个字符，从左上到右下构成 2020。

例如，对于下面的矩阵：
220000
000000
002202
000000
000022
002020
一共有 5 个 2020。其中 1 个是在同一行里的，1 个是在同一列里的，3 个是斜线上的。
小蓝的矩阵比上面的矩阵要大，由于太大了，他只好将这个矩阵放在了一个文件里面，在试题目录下有一个文件 2020.txt，里面给出了小蓝的矩阵。
请帮助小蓝确定在他的矩阵中有多少个 2020。

思路代码

直译

a = []
with open(r'C:\Users\76390\Desktop\2020.txt', 'r') as fp:
    for line in fp.readlines():
        a.append(line.strip())
m = len(a)
n = len(a[0])
num = 0
# 横向寻找
for i in range(m):
    for j in range(n-3):
        if a[i][j] == '2' and a[i][j+1] == '0' and a[i][j+2] == '2' and a[i][j+3] == '0':
            num += 1
# 纵向寻找
for b in range(m-3):
    for c in range(n):
        if a[b][c] == '2' and a[b+1][c] == '0' and a[b+2][c] == '2' and a[b+3][c] == '0':
            num += 1
# 斜寻找
for l in range(m-3):
    for k in range(n-3):
        if a[l][k] == '2' and a[l+1][k+1] == '0' and a[l+2][k+2] == '2' and a[l+3][k+3] == '0':
            num += 1

print(num)