题目:
给你一个由一些多米诺骨牌组成的列表 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) 的数量。
思路:
暴力求解会超时。该如何处理?
1.对每张牌,从后向前遍历,统计等价牌的数量,判断等价后,将其从数组中删除。
2.等价牌的对数:C(n,2)=n*(n-1)/2
解答:
class Solution:
def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
n=len(dominoes)
res=0
for i in range(len(dominoes)):
#tmp用来记录与当前牌等价的牌的数量
tmp=1
for j in range(len(dominoes)-1,i,-1):
if (dominoes[i][0]==dominoes[j][0] and dominoes[i][1]==dominoes[j][1]) or (dominoes[i][0]==dominoes[j][1] and dominoes[i][1]==dominoes[j][0]):
tmp+=1
del dominoes[j]
if tmp>1:
res+=tmp*(tmp-1)//2
return res