题目描述:给你一个由一些多米诺骨牌组成的列表 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) 的数量。
解题思路一:现将所有的多米诺骨牌都转化为小的在前,大的在后,然后对数组进行排序,这样等价的多米诺骨牌都会放在一块,统计每个不同的多米诺骨牌出现的次数即可得到等价对的数量,代码如下:
class Solution:
def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
n = len(dominoes)
for i in range(n):
dominoes[i] = [min(dominoes[i]), max(dominoes[i])]
dominoes.sort()
res = 0
pre = dominoes[0]
count = 1
for i in range(1, n):
if dominoes[i] == pre:
count += 1
else:
res += count * (count-1) // 2
count = 1
pre = dominoes[i]
res += count * (count -1) // 2
return res
解题思路二:将多米诺骨牌按照小的在左大的在右的顺序表示为0~99之间的一个数,然后统计每个数次出现的次数,即可得到等价对的数量,代码如下:
class Solution:
def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
res = 0
count = [0] * 100
for do in dominoes:
val = min(do[0]*10 + do[1], do[1]*10+do[0])
res += count[val]
count[val] += 1
return res