yankewei / LeetCode

LeetCode 问题的解决方法
MIT License
6 stars 0 forks source link

等价多米诺骨牌对的数量 #100

Open yankewei opened 3 years ago

yankewei commented 3 years ago

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

提示:

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

yankewei commented 3 years ago

哈希表

很简单,只要把等价的两个数组用一个字符串或者int表示就可以了。因为已经确定都是1到9的数字,那么我们可以使用int来表示。

func numEquivDominoPairs(dominoes [][]int) int {
    var ret int
    m := map[int]int{}
    for i := 0; i < len(dominoes); i++ {
    a, b := dominoes[i][0], dominoes[i][1]
    var value int
    if a > b {
        value = a*10 + b
    } else {
        value = b*10 + a
    }
        // 这里要先计算数量再加
    if _, e := m[value]; e {
        ret += m[value]
    }
    m[value]++
    }
    return ret
}