LeetCode1128-等价多米诺骨牌对的数量

发表于 | 分类于

题目链接

英文链接：https://leetcode.com/problems/number-of-equivalent-domino-pairs/

中文链接：https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/

题目详述

给你一个由一些多米诺骨牌组成的列表 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
输入：dominoes = [[1,2],[2,1],[3,4],[5,6]]
输出：1

提示：

  • 1 <= dominoes.length <= 40000
  • 1 <= dominoes[i][j] <= 9

题目详解

  • 运用哈希表统计等价多米诺骨牌对的数量，取 C(n, 2) 即可。
  • 哈希表的键与两个值都有关，由于 1 <= dominoes[i][j] <= 9，可以把两个数转化成一个数作为键（比如采取小的在前、大的在后这种方式）。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
public class LeetCode_01128 {

    public int numEquivDominoPairs(int[][] dominoes) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int[] d : dominoes) {
            int x = d[0] < d[1] ? d[0] * 10 + d[1] : d[1] * 10 + d[0];
            map.put(x, map.getOrDefault(x, 0) + 1);
        }
        int res = 0;
        for (int v : map.values()) {
            res += v * (v - 1) / 2;
        }
        return res;
    }
}