【问题描述】
小蓝有一个数字矩阵,里面只包含数字 0 和 2。小蓝很喜欢 2020,他想找
到这个数字矩阵中有多少个 2020 。
小蓝只关注三种构成 2020 的方式:
• 同一行里面连续四个字符从左到右构成 2020。
• 同一列里面连续四个字符从上到下构成 2020。
• 在一条从左上到右下的斜线上连续四个字符,从左上到右下构成 2020。
例如,对于下面的矩阵:
220000
000000
002202
000000
000022
002020
一共有 5 个 2020。其中 1 个是在同一行里的,1 个是在同一列里的,3 个
是斜线上的。
小蓝的矩阵比上面的矩阵要大,由于太大了,他只好将这个矩阵放在了一个文件里面,在试题目录下有一个文件 2020.txt,里面给出了小蓝的矩阵。
请帮助小蓝确定在他的矩阵中有多少个 2020。
nums = [[2, 2, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0],
[0, 0, 2, 2, 0, 2],
[0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 2, 2],
[0, 0, 2, 0, 2, 0],
]
n = 0
for i in range(len(nums)):
for j in range(len(nums[0])):
if j+4 <= len(nums[0]): # 行中2020
a = nums[i][j:j+4]
if a == [2,0,2,0]:
n+=1
if i+3<len(nums): # 列中出现2020
b = nums[i][j]*1000+nums[i+1][j]*100+nums[i+2][j]*10+nums[i+3][j]
if b == 2020:
n+=1
if i+3<len(nums) and j+3 < len(nums[0]): # 从左上到右下出现2020
c = nums[i][j]*1000+nums[i+1][j+1]*100+nums[i+2][j+2]*10+nums[i+3][j+3]
if c == 2020:
n+=1
print(n)