给你一个由一些多米诺骨牌组成的列表 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
C++
class Solution {
public:
int numEquivDominoPairs(vector<vector<int>>& dominoes)
{
for(int i=0;i<dominoes.size();i++)
{
sort(dominoes[i].begin(),dominoes[i].end());
}
map<vector<int>,int> tmp;
int res=0;
for(auto it:dominoes)
{
tmp[it]++;
}
for(auto it:tmp)
{
res+=it.second*(it.second-1)/2;
}
return res;
}
};
python
class Solution:
def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
for it in dominoes:
it.sort()
dic={}
for it in dominoes:
it=str(it)
if it not in dic:
dic[it]=1
else:
dic[it]+=1
res=0
for key in dic:
res+=dic[key]*(dic[key]-1)//2
return res