给你一个由一些多米诺骨牌组成的列表 dominoes。
如果其中某一张多米诺骨牌可以通过旋转 0 度或 180 度得到另一张多米诺骨牌,我们就认为这两张牌是等价的。
形式上,dominoes[i] = [a, b] 和 dominoes[j] = [c, d] 等价的前提是 ac 且 bd,或是 ad 且 bc。
在 0 <= i < j < dominoes.length 的前提下,找出满足 dominoes[i] 和 dominoes[j] 等价的骨牌对 (i, j) 的数量。
示例:
输入:dominoes = [[1,2],[2,1],[3,4],[5,6]]
输出:1
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs
思路:
首先,我们将多米诺骨牌堆中对应的牌进行相互转换,例如将牌(1,2),(2,1)均转换成(1,2),方便计数,可对每个牌用sort方法处理。
再对整个牌堆排序,使相同的牌排在一起。
牌数初始化为1,遍历牌堆进行计数,每次加相同牌数的阶乘,直至遍历结束。
注意的是,遍历结束后仍有最后一组牌未进行处理,需在循环外处理
class Solution:
def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
for i in dominoes:
i.sort()
dominoes.sort()
res,cnt = 0,1
for p in range(len(dominoes)-1):
if dominoes[p] == dominoes[p+1]:
cnt += 1
else:
res += cnt*(cnt-1)//2
cnt = 1
return res + cnt*(cnt-1) // 2