3234 蓝桥杯2024年第十五届省赛真题-连连看

仅记录

m,n=map(int,input().split())

a=[list(map(int,input().split())) for _ in range(m)]

str1=[[-1]*2000 for _ in range(2000)]

str2=[[-1]*2000 for _ in range(2000)]

for i in range(m):

    for j in range(n):

        str1[i-j+1000][a[i][j]]+=1

        str2[i+j][a[i][j]]+=1

num=0

for i in range(m):

    for j in range(n):

        num=num+str1[i-j+1000][a[i][j]]

        num=num+str2[i+j][a[i][j]]

print(num)