leetCode1128:多米诺骨牌的等价对数，自定义哈希函数

目录

一、题目描述

二、解题思路

三、代码实现

---

一、题目描述

给你一个由一些多米诺骨牌组成的列表 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是pair类型，需要自定义pair类型的哈希函数；
• 为了方便处理，可以将两种等价形式转换成一种，只需要把数对中的数统一成左边的数小于等于右边的数即可

三、代码实现

#include <bits/stdc++.h>
using namespace std;

//构建pair类型的哈希函数
struct hash_pair {
    template <class T1, class T2>
    size_t operator()(const pair<T1, T2>& p) const {
        auto hash1 = hash<T1>{}(p.first);
        auto hash2 = hash<T2>{}(p.second);
        return hash1 ^ hash2;
    }
};
int numEquivDominoPairs(vector<vector<int>>& dominoes) {
    int n = dominoes.size();
    unordered_map<pair<int, int>, int, hash_pair> cnt;
    int res = 0;
    //先统一等价的形式：
    //dominoes[i] = [a, b] 和 dominoes[j] = [c, d] 等价的是 a==c 且 b==d
    for (int i = 0; i < n; i++) {
        if (dominoes[i][0] > dominoes[i][1]) {
            int tmp = dominoes[i][0];
            dominoes[i][0] = dominoes[i][1];
            dominoes[i][1] = tmp;
        }
    }
    //统计等价的数量
    for (int i = 0; i < n; i++) {
        if (cnt.count({dominoes[i][0], dominoes[i][1]})) {
            res += cnt[{dominoes[i][0], dominoes[i][1]}];
        }
        cnt[{dominoes[i][0], dominoes[i][1]}]++;
    }
    return res;
}
int main() {
    vector<vector<int>> dominoes = {{1, 2}, {1, 2}, {1, 1}, {1, 2}, {2, 2}};

    cout << numEquivDominoPairs(dominoes);
    return 0;
}