[蓝桥杯python实现寻找2020]

自己写了一个比较简单的分享一下，有不足之处还希望大佬指正！

with open('test.txt', 'r', encoding='utf-8') as p:
    lines = p.readlines()
list_1 = []
for i in lines:
    list_1.append(i.strip())
print(list_1)
print(len(list_1), len(list_1[0]))
count = 0
for i in range(300):
    for p in range(300):
        if i+3 <= 299 and p+3 <= 299 and (list_1[i][p] + list_1[i + 1][p + 1] + list_1[i + 2][p + 2] + list_1[i + 3][p + 3] == '2020')  :
            count +=1
        if p+3 <= 299 and (list_1[i][p]+list_1[i][p + 1] + list_1[i][p + 2] + list_1[i][p + 3] == '2020') :
            count += 1
        if i+3 <= 299 and (list_1[i][p] + list_1[i + 1][p] +list_1[i + 2][p] + list_1[i + 3][p] == '2020') :
            count += 1
print(count)