题目描述
给你一个由一些多米诺骨牌组成的列表 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
(map排列组合) $O(n)$
用一个map来存多米诺骨牌对,key值存的是多米诺骨牌对(ex: [3,4]),而且这个多米诺骨牌对较小的数在前,如果两个多米诺牌对key值相等,那么等价(ex:[3,4]和[4,3])。
value值用来存这一个等价多米诺骨牌对出现的次数,最后通过排列组合公式就可以算出总共等价的多米诺骨牌对个数。
C++ 代码
class Solution {
public:
int numEquivDominoPairs(vector<vector<int>>& d) {
map<pair<int,int>, int> cnt;
int n = d.size();
for (int i = 0; i < n; ++ i )
{
int first = min(d[i][0], d[i][1]), second = max(d[i][0], d[i][1]);
cnt[{first, second}]++;
}
int ans = 0;
for (auto x : cnt)
{
int nums = x.second;
ans += nums * (nums-1) / 2;
}
return ans;
}
};