【蓝桥杯Python组】寻找2020

寻找2020

【题目描述】

小蓝有一个数字矩阵，里面只包含数字 0 和 2。小蓝很喜欢 2020，他想找到这个数字矩阵中有多少个 2020 。小蓝只关注三种构成 2020 的方式：
• 同一行里面连续四个字符从左到右构成 2020。
• 同一列里面连续四个字符从上到下构成 2020。
• 在一条从左上到右下的斜线上连续四个字符，从左上到右下构成 2020。

【样例输入】

220000
000000
002202
000000
000022
002020
202020

【样例输出】

7

【暴力求解】
求解过程中只需要区分好行列之间的关系即可，无思维含量

content = []

with open('LanQiao.txt') as fp:
	for line in fp.readlines():
		content.append(list(line.strip()))

# 初始化行列值与总共的2020数量
row=len(content)
col=len(line)
sum=0

# 横向遍历
for i in range(row):
	for j in range(col-3):
		if content[i][j]=="2" and content[i][j+1]=="0" and content[i][j+2]=="2" and content[i][j+3]=="0":
			sum+=1

# 纵向遍历
for i in range(row-3):
	for j in range(col):
		if content[i][j]=="2" and content[i+1][j]=="0" and content[i+2][j]=="2" and content[i+3][j]=="0":
			sum+=1

# 斜向遍历
for i in range(row-3):
	for j in range(col-3):
		if content[i][j]=="2" and content[i+1][j+1]=="0" and content[i+2][j+2]=="2" and content[i+3][j+3]=="0":
			sum+=1

print(sum)