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

源码:

class Solution {
     
    public int numEquivDominoPairs(int[][] dominoes) {
     
        Map<String, Integer> map = new HashMap<>();
        int count = 0;
        for (int i = 0; i < dominoes.length; i++) {
     
            // 利用字符串来区别
            // 注意:此处已经将等价和相同的情况都列入
            // 按照大小排序好了,所以之后的 str 比较时就不用考虑等价和相同两种情况
            String str = dominoes[i][1] > dominoes[i][0] ? 
                dominoes[i][0] + "" + dominoes[i][1] :
                dominoes[i][1] + "" + dominoes[i][0];
            if (map.containsKey(str)) {
     
                map.put(str, map.get(str) + 1);
                // 因为之前如果存在一对,再来一个相似的
                // 那么就是三对多米诺骨牌
                // 所以加的数量是添加 1 后的map里的多米诺骨牌数量
                count += map.get(str);
            } else {
     
                map.put(str, 0);
            }
        }
        return count;
    }
}

你可能感兴趣的:(LeetCode)