[蓝桥杯2020初赛] 寻找2020Python版

题目描述

小蓝有一个数字矩阵，里面只包含数字0 和2。小蓝很喜欢2020，他想找到这个数字矩阵中有多少个2020 。
小蓝只关注三种构成2020 的方式：
同一行里面连续四个字符从左到右构成2020。
同一列里面连续四个字符从上到下构成2020。
在一条从左上到右下的斜线上连续四个字符，从左上到右下构成2020。
例如，对于下面的矩阵：
220000
000000
002202
000000
000022
002020
一共有5 个2020。其中1 个是在同一行里的，1 个是在同一列里的，3 个是斜线上的。 小蓝的矩阵比上面的矩阵要大，由于太大了，他只好将这个矩阵放在了一个文件里面。
在本试题中有一个文件2020.txt，里面给出了小蓝的矩阵。
请帮助小蓝确定在他的矩阵中有多少个2020。
这是一道结果填空的题，你只需要算出结果后提交即可。
本题的结果为一个整数，在提交答案时只填写这个整数，填写多余的内容将无法得分。

#[蓝桥杯2020初赛] 寻找2020Python版
matrix=[]
with open("2020.txt") as fp:
    for a in fp.readlines():
        matrix.append(list(a.strip()))
row = len(matrix)  
col = len(matrix[0])
count = 0
for i in range(row):
    for j in range(col):
        if matrix[i][j] == '2':  
            if j + 3 < col:
                if matrix[i][j] + matrix[i][j + 1] + matrix[i][j + 2] + matrix[i][j + 3] == "2020":
                    count += 1
            if i + 3 < row:
                if matrix[i][j] + matrix[i + 1][j] + matrix[i + 2][j] + matrix[i + 3][j] == "2020":
                    count += 1
            if i + 3 < row and j + 3 < col:
                if matrix[i][j] + matrix[i + 1][j + 1] + matrix[i + 2][j + 2] + matrix[i + 3][j + 3] == "2020":
                    count += 1
print(count)