每日1题 -相等行列对

每日1题 -相等行列对_第1张图片

每日1题 -相等行列对_第2张图片 每日1题 -相等行列对_第3张图片


 我的代码↓

class Solution:
    def equalPairs(self, grid: List[List[int]]) -> int:

        ans=0

        n=len(grid)

        for row in range(n):
            r=grid[row]

            for col in range(n):
                flag=1

                for c in range(n):
                    if grid[c][col]==r[c]:
                        continue
                    else:
                        flag=0
                        break

                if flag:
                    ans+=1
        
        return ans            

官方题解:哈希表

class Solution:
    def equalPairs(self, grid: List[List[int]]) -> int:
        res, n = 0, len(grid)
        cnt = Counter(tuple(row) for row in grid)
        res = 0
        for j in range(n):
            res += cnt[tuple([grid[i][j] for i in range(n)])]
        return res

Counter() 是 collections 库中的一个函数,可以用来统计一个 python 列表、字符串、元组等可迭代对象中每个元素出现的次数,并返回一个字典

你可能感兴趣的:(LeetCode算法,leetcode,算法,python)