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

Leetcode 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) 的数量。

示例：

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

思路

• 首先要看懂题目的意思， 旋转180指的是原来的牌[1,2], 旋转后成为[2,1]
• 依次遍历每一位， 计算出对应的二位数的和， 再与之前已经记录在map之中的个数相加即可
• 注意如果是[1,1]这样的， 由于旋转180和0都是二位数为11， 所以这种情况要特判

代码 —— golang

func numEquivDominoPairs(dominoes [][]int) int {
	count, res := make(map[int]int), 0

	for i := 0;i < len(dominoes);i++ {
		data := dominoes[i][0] * 10 + dominoes[i][1]
		data_rev := dominoes[i][1] * 10 + dominoes[i][0]
	
		if data != data_rev {
			res += count[data] + count[data_rev]
		} else {
			res += count[data]
		}
		
		count[data] += 1
	}

	return res
}