天天看点

1128. Number of Equivalent Domino Pairs*

1128. Number of Equivalent Domino Pairs*

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

题目描述

Given a list of ​

​dominoes​

​​, ​

​dominoes[i] = [a, b]​

​​ is equivalent to ​

​dominoes[j] = [c, d]​

​​ if and only if either ​

​(a==c and b==d)​

​​, or ​

​(a==d and b==c)​

​ - that is, one domino can be rotated to be equal to another domino.

Return the number of ​

​pairs (i, j)​

​​ for which ​

​0 <= i < j < dominoes.length​

​​, and ​

​dominoes[i]​

​​ is equivalent to ​

​dominoes[j]​

​.

Example 1:

Input: dominoes = [[1,2],[2,1],[3,4],[5,6]]
Output: 1      

Constraints:

  • ​1 <= dominoes.length <= 40000​

  • ​1 <= dominoes[i][j] <= 9​

C++ 实现 1

采用 1. Two Sum*​ 的思路, 利用一个哈希表保存已经访问过的 ​

​domino​

​​. 这题容易需要注意的地方是, 比如 ​

​dominoes = [[1, 2], [1, 2], [1, 2], [1, 2]]​

​​, 最后返回的结果应该是 ​

​3 + 2 + 1 = 6​

​.

class Solution {
private:
    struct HashVector {
        std::size_t operator()(const std::vector<int> &vec) const {
            std::size_t seed = vec.size();
            for(auto& i : vec) {
                seed ^= i + 0x9e3779b9 + (seed << 6) + (seed >> 2);
             }
            return seed;
        }
    };
public:
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
        unordered_map<vector<int>, int, HashVector> records;
        int res = 0;
        for (auto &d : dominoes) {
            // 使最小值放在前面
            std::sort(d.begin(), d.end());
            if (records.count(d)) res += records[d] ++;
            else records[d] ++;
        }
        return res;
    }
};