leetcode_1128. 等价多米诺骨牌对的数量

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给你一个由一些多米诺骨牌组成的列表 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

二、解题思路

双向选择,可以认为旋转后一致的为同一个key,那么我们只需要知道个数大于1的就行,然后从这里面任选2个即可,因此任选2个的组合就是每组等价牌的数量,每组等价牌的组合加起来即可。

三、代码

class Solution:
    def numEquivDominoPairs(self, dominoes: list) -> int:
        num_dict = {}
        for i in range(len(dominoes)):
            if (dominoes[i][1], dominoes[i][0]) in num_dict:
                num_dict[(dominoes[i][1], dominoes[i][0])] += 1
            elif (dominoes[i][0], dominoes[i][1]) in num_dict:
                num_dict[(dominoes[i][0], dominoes[i][1])] += 1
            elif (dominoes[i][0], dominoes[i][1]) not in num_dict:
                num_dict[(dominoes[i][0], dominoes[i][1])] = 1

        def bilateral_selection(n):
            return int(n * (n - 1) / 2)

        count = 0
        for num in num_dict:
            if num_dict[num] > 1:
                count += bilateral_selection(num_dict[num])
        return count


if __name__ == '__main__':
    s = Solution()
    dominoes = [[1, 2], [2, 1], [3, 4], [5, 6]]
    ans = s.numEquivDominoPairs(dominoes)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
相关推荐
©️2020 CSDN 皮肤主题: 博客之星2020 设计师:CY__ 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值