问题描述
给你一个由一些多米诺骨牌组成的列表 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析及代码实现
将列表中的每个元素进行排序,然后转换成不可变类型,在这里我转的是元组,存放到字典中,如果有相同的元素,那么个数+1,最后遍历字典,求所有对的个数,每种有对子的元素,所对应总的对子个数是一个等差数,可以用n*(n-1)/2求出他的总和,因为最后输出的是没有小数点的整数,所以要用整除//
class Solution:
def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
ans = 0
dic = {}
for m, n in dominoes:
index = tuple(sorted((m, n)))
if index in dic:
dic[index] += 1
else:
dic[index] = 1
for i in dic:
if dic[i] > 1:
ans += dic[i] * (dic[i]-1) // 2
return ans