1128. 等价多米诺骨牌对的数量

给你一个由一些多米诺骨牌组成的列表 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) 的数量

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

示例

输入:dominoes = [[1,2],[2,1],[3,4],[5,6]]
输出:1

提示:

  • ​1 <= dominoes.length <= 40000​
  • ​1 <= dominoes[i][j] <= 9​

思路

先统计出每个牌的数,记下他们的次数。那么这个数的牌就是等价的。遍历存放次数的容器,每一个数的对数就是map[i]*(map[i]-1)/2;把总的加起来返回即可。

代码

class Solution {
public:
int numEquivDominoPairs(vector<vector<int>>& dominoes) {
vector<int> map(100);
int res=0;
for(int i=0;i<dominoes.size();i++)
{
int m=dominoes[i][0],n=dominoes[i][1];
int k=m>n?m*10+n:n*10+m;
map[k]++;//次数+1
}
for(int i=0;i<map.size();i++)
{
res+=map[i]*(map[i]-1)/2;
}
return res;
}
};