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))
测试矩阵结果
文件结果