链接
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
C_t^2
Ct2,最后求和所有种等价骨牌对的数量即是答案。计数每种等价骨牌的数量,首先将所有 a > b 都调换位置,变成前小后大的形式,这样等价的前提变成只有 a==c 且 b==d;然后根据提示知 1 <= dominoes[i][j] <= 9
并且每张骨牌只有两位,所以第一位乘十加第二位(a*10+b)即可唯一表示所有情况。用(a*10+b)作为哈希键值计数每种等价骨牌的数量。
时间复杂度: O ( n ) O(n) 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;
}
};