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
}