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

20210126 每日一题
题面:
给你一个由一些多米诺骨牌组成的列表 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) 的数量。

思路:很简单,考虑用map就好了。无论怎样只维护<a,b>出现的次数(这里统一a<b,如果不符合,交换一下记录就好了)。然后最后考虑次数超过1的pair,最简单的组合数n个里挑2个就好了。

代码:

class Solution {
    
    
public:
    map<pair<int,int>,int>mp;
    int num[100];
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
    
    
            int cnt=0;
            int idx=1;
            for(int i=0;i<dominoes.size();++i){
    
    
                int a = dominoes[i][0];
                int b = dominoes[i][1];
                if(a>b) swap(a,b);
                pair<int,int> p1 = pair(a,b);
                if(mp[p1]==0) {
    
    num[idx]=1;mp[p1]=idx++;} 
                else{
    
    
                    int tidx = mp[p1];
                    num[tidx]++;
                    if(num[tidx]==2) cnt++;
                }
            }
            int ans=0;
            for(int i=1;i<idx;++i)
                if(num[i]>1) ans+=(num[i]-1)*num[i]/2;
            return ans;
    }
};

每日记事:早起打卡成功,吃碗焖肉面庆祝一下!

猜你喜欢

转载自blog.csdn.net/weixin_43329358/article/details/113172173