【陣列】1128. 等價多米諾骨牌對的數量(簡單)

小學三年級01班發表於2020-11-04

【題目】
給你一個由一些多米諾骨牌組成的列表 dominoes。如果其中某一張多米諾骨牌可以通過旋轉 0 度或 180 度得到另一張多米諾骨牌,我們就認為這兩張牌是等價的。形式上,dominoes[i] = [a, b] 和 dominoes[j] = [c, d] 等價的前提是 ac 且 bd,或是 ad 且 bc。在 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
【程式碼】

class Solution {
public:
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
        map<int,int> m;
        int cnt=0,num1,num2;
        for(auto x:dominoes){
            num1=x[0]*10+x[1];
            num2=x[1]*10+x[0];
            if(num1<num2)
                m[num1]++;
            else
                m[num2]++;
        }
        for(auto x:m)
            cnt+=x.second*(x.second-1)/2;
        return cnt;
    }
};

相關文章