11.python解答2020年蓝桥杯省赛python组 寻找2020

11.python解答2020年蓝桥杯省赛python组 寻找2020

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

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

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

2020.txt下载https://wwe.lanzous.com/iicFtmyxd7e

思路

暴力法，循环枚举每一个矩阵坐标，看坐标三个方向的四位数是否组成2020

import time
time1 = time.time()

mat = []
mat = ['220000',
'000000',
'002202',
'000000',
'000022',
'002020']

# 如果用文件的2020，请去掉注释，同时把上面的mat矩阵注释掉
#with open(r'C:\Users\mgboy\Desktop\2020.txt','r',encoding='utf-8') as f:
    #for line in f.readlines():
        #mat.append(line.strip())


m = len(mat)# 行数
n = len(mat[0])# 列数
print('m = {} n = {}'.format(m,n))

num = '2020'

count = 0

for i in range(m):
    for j in list(range(n)):
        # 横向
        if j+3 < n:
            # 横向
            if mat[i][j:j+4] == '2020':
                count += 1
        
        # 竖向
        if i+3 < m:
            s = ''
            for ii in list(range(i,i+4)):
                s += mat[ii][j]
            if s == '2020':
                count += 1
        # 斜向
        if j+3 < n and i+3 < m:
            s = ''
            for ii,jj in zip(list(range(i,i+4)), list(range(j, j+4))):
                s += mat[ii][jj]
            if s == '2020':
                count += 1

print(count)

print('花费时间：{}'.format(time.time()-time1))

测试矩阵结果

文件结果