【leetcode】1128. 等价多米诺骨牌对的数量（number-of-equivalent-domino-pairs）（哈希）[简单]

链接

https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/

耗时

解题：50 min
题解：16 min

题意

给你一个由一些多米诺骨牌组成的列表 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 <= dominoes.length <= 40000
• 1 <= dominoes[i][j] <= 9

思路

找到所有情况下的等价的骨牌的数量，对于每种等价的骨牌的数量 t 的等价骨牌对数即是 $C_t^2$，最后求和所有种等价骨牌对的数量即是答案。计数每种等价骨牌的数量，首先将所有 a > b 都调换位置，变成前小后大的形式，这样等价的前提变成只有 a==c 且 b==d；然后根据提示知 1 <= dominoes[i][j] <= 9 并且每张骨牌只有两位，所以第一位乘十加第二位（a*10+b）即可唯一表示所有情况。用（a*10+b）作为哈希键值计数每种等价骨牌的数量。

时间复杂度：$O(n)$

AC代码

class Solution {
public:
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
        unordered_map<int, int> docnt;
        for(auto domino : dominoes) {
            if(domino[1] < domino[0]) {
                swap(domino[0], domino[1]);
            }
            int index = domino[1]*10+domino[0];
            if(docnt.find(index) != docnt.end()) {
                docnt[index]++;
            }
            else {
                docnt[index] = 1;
            }
        }
        int ans = 0;
        for(auto x : docnt) {
            int cnt = x.second;
            ans += cnt*(cnt-1)/2;
        }
        return ans;
    }
};