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)