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

2020-02-20  本文已影响0人  周英杰Anita

题目描述:

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

思路:

遍历,先排序,消除等价骨牌差异性,然后每多一个A骨牌的等价骨牌,给当前对数 贡献了目前A骨牌的个数

Java解法:

class Solution {
    public int numEquivDominoPairs(int[][] dominoes) {
        //获取数组的列数
        int ans = 0;
        int[] tmp = new int[100];
        for (int[] arr : dominoes){
            Arrays.sort(arr);
            ans += tmp[arr[0] * 10 + arr[1]] ++;
        }
        return ans;
    }
}

python3解法1:

lass Solution:
    def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
        ans = 0;
        d = dict()
        for d1, d2 in dominoes:
            tmp = tuple(sorted((d1, d2)))
            if tmp in d:
                d[tmp] += 1
            else:
                d[tmp] = 1
        for i in d:
            ans += d[i] * (d[i] - 1) // 2
        return ans

python3解法2:

class Solution:
    def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
        tmp = {}
        ans = 0
        for d1, d2 in dominoes:
            if d1 > d2:
                d1, d2 = d2, d1 
            num = d1 * 10 + d2
            if num not in tmp:
                tmp[num] = 0
            ans += tmp[num]
            tmp[num] +=1
        return res

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs

上一篇下一篇

猜你喜欢

热点阅读