【Leetcode_总结】 1128. 等价多米诺骨牌对的数量 - python

Q:

给你一个由一些多米诺骨牌组成的列表 dominoes。

如果其中某一张多米诺骨牌可以通过旋转 0 度或 180 度得到另一张多米诺骨牌,我们就认为这两张牌是等价的。

形式上,dominoes[i] = [a, b] 和 dominoes[j] = [c, d] 等价的前提是 a==c 且 b==d,或是 a==d 且 b==c。

在 0 <= i < j < dominoes.length 的前提下,找出满足 dominoes[i] 和 dominoes[j] 等价的骨牌对 (i, j) 的数量。

 

示例:

输入:dominoes = [[1,2],[2,1],[3,4],[5,6]]
输出:1
 

提示:

1 <= dominoes.length <= 40000
1 <= dominoes[i][j] <= 9


思路:

链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/

代码:

class Solution:
    def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
        res = 0
        dic = {}
        for i in range(len(dominoes)):
            if str(dominoes[i]) in dic:
                dic[str(dominoes[i])] += 1
            elif str(dominoes[i][::-1]) in dic:
                dic[str(dominoes[i][::-1])] += 1
            else:
                dic[str(dominoes[i])] = 1
        for k, v in dic.items():
            if v > 1:
                res += v*(v-1)//2
        return res

【Leetcode_总结】 1128. 等价多米诺骨牌对的数量 - python_第1张图片

你可能感兴趣的:(Leetcode,数组)