README.md

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

English Version

题目描述

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

解法

Python3

class Solution:
    def numEquivDominoPairs(self, dominoes: List[List[int]]) -> int:
        counter = Counter()
        ans = 0
        for a, b in dominoes:
            v = a * 10 + b if a > b else b * 10 + a
            ans += counter[v]
            counter[v] += 1
        return ans

Java

class Solution {
    public int numEquivDominoPairs(int[][] dominoes) {
        int ans = 0;
        int[] counter = new int[100];
        for (int[] d : dominoes) {
            int v = d[0] > d[1] ? d[0] * 10 + d[1] : d[1] * 10 + d[0];
            ans += counter[v];
            ++counter[v];
        }
        return ans;
    }
}

class Solution {
    public int numEquivDominoPairs(int[][] dominoes) {
        int[] counter = new int[100];
        for (int[] d : dominoes) {
            int v = d[0] > d[1] ? d[0] * 10 + d[1] : d[1] * 10 + d[0];
            ++counter[v];
        }
        int ans = 0;
        for (int c : counter) {
            ans += c * (c - 1) / 2;
        }
        return ans;
    }
}

C++

class Solution {
public:
    int numEquivDominoPairs(vector<vector<int>>& dominoes) {
        vector<int> counter(100);
        int ans = 0;
        for (auto& d : dominoes)
        {
            int v = d[0] > d[1] ? d[0] * 10 + d[1] : d[1] * 10 + d[0];
            ans += counter[v];
            ++counter[v];
        }
        return ans;
    }
};

Go

func numEquivDominoPairs(dominoes [][]int) int {
	counter := make([]int, 100)
	for _, d := range dominoes {
		if d[1] < d[0] {
			d[0], d[1] = d[1], d[0]
		}
		v := d[0]*10 + d[1]
		counter[v]++
	}
	ans := 0
	for _, c := range counter {
		ans += c * (c - 1) / 2
	}
	return ans
}

...