【Python】2020年蓝桥杯省赛真题——寻找2020

问题描述

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

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

  例如，对于下面的矩阵：
    220000
    000000
    002202
    000000
    000022
    002020
  一共有 5 个 2020。其中 1 个是在同一行里的，1 个是在同一列里的，3 个是斜线上的。

  小蓝的矩阵比上面的矩阵要大，由于太大了，他只好将这个矩阵放在了一个文件里面，在试题目录下有一个文件 2020.txt，里面给出了小蓝的矩阵。

  请帮助小蓝确定在他的矩阵中有多少个 2020。

  链接: https://pan.baidu.com/s/1mteb3fcd6Bjf5Jb3jYMFwA
  提取码: av8i

代码实现

"""实现想法：将文件读入为字符串列表后拆分为字符列表，以字符为单位逐位对比"""
count = 0
contents = []

filename = '2020.txt'
with open(filename) as f:
    for line in f.readlines():
        contents.append(list(line.strip()))

# Horizontal search
for i in range(len(contents)):
    for j in range(len(contents[i]) - 3):
        if contents[i][j] == '2' and contents[i][j+1] == '0'\
                and contents[i][j+2] == '2' and contents[i][j+3] == '0':
            count += 1

# Vertical search
for i in range(len(contents) - 3):
    for j in range(len(contents[i])):
        if contents[i][j] == '2' and contents[i+1][j] == '0'\
                and contents[i+2][j] == '2' and contents[i+3][j] == '0':
            count += 1

# Diagonal search
for i in range(len(contents) - 3):
    for j in range(len(contents[i]) - 3):
        if contents[i][j] == '2' and contents[i+1][j+1] == '0'\
                and contents[i+2][j+2] =='2' and contents[i+3][j+3] == '0':
            count += 1

print(count)
------------------------------------------------------------------------------------
Output: 16520