连连看，我看不懂啊啊啊

import os
import sys
from collections import defaultdict
# 请在此输入您的代码
n,m = map(int,input().split())
num1 = defaultdict(lambda:defaultdict(int))
num2 = defaultdict(lambda:defaultdict(int))
num = []
for i in range(n):
    num.append(list(map(int,input().split())))
for i in range(n):
    for j in range(m):
        num1[num[i][j]][i-j] += 1
        num2[num[i][j]][i+j] += 1
res = 0
for i in num1:
    for j in num1[i]:
        res += num1[i][j]*(num1[i][j] - 1)
for i in num2:
    for j in num2[i]:
        res += num2[i][j]*(num2[i][j] - 1)
print(res)

0连连看 - 蓝桥云课

使用字典，value也是字典，因为abs(a-b) == abs(c-d) and >0，说明相等的元素只能在对角线或者副对角线上，字典的键是元素的值，字典的值是一个字典，这个字典中的值是主对角线上上坐标的差，或者副对角线上坐标的和，value是相等元素的个数。当有n个元素相等时，坐标对应该有n*（n-1）对。