连连看题解

解题思路:

注意事项:

参考代码:

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

N = int(2e3+5)

st1 = [[0] * N for _ in range(N)]

st2 = [[0] * N for _ in range(N)]

arr = []

for _ in range(n):

    arr += [list(map(int, input().split()))]

for i in range(n):

    for j in range(m):

        st1[i - j + 1000][arr[i][j]] += 1

        st2[i + j][arr[i][j]] += 1

ans = 0

for i in range(n):

    for j in range(m):

        ans += st1[i - j + 1000][arr[i][j]] - 1

        ans += st2[i + j][arr[i][j]] - 1

print(ans)