LeetCode 1128. 等价多米诺骨牌对的数量(简单)

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

 

提示:

  • 1 <= dominoes.length <= 40000
  • 1 <= dominoes[i][j] <= 9

我的Java代码:

思路:一开始是直接嵌套循环暴力求解,然后超时。想着遍历到第一个有等价牌的骨牌时,里面的循环就已经把它等价的牌都找到了,用公式即可求出这组等价牌两两等价的对数,就不用再去遍历这组牌中的其他牌了,因此考虑用一个访问数组,来标记已访问的牌,减少内层循环的次数,减少运行时间。最后虽然过了,但是效率还是不高。

class Solution {
    public int numEquivDominoPairs(int[][] dominoes) {
        int len = dominoes.length;
		boolean[] visited = new boolean[len];
		int sum = 0;
		for(int i = 0;i < len;i++) {
			if(visited[i]) {
				continue;
			}
			int num = 1;
			visited[i] = true;
			for(int j = i+1;j < len;j++) {
				if(visited[j]) {
					continue;
				}
				if(dominoes[i][0] == dominoes[j][0] && dominoes[i][1] == dominoes[j][1]
						|| dominoes[i][0] == dominoes[j][1] && dominoes[i][1] == dominoes[j][0]) {
					num++;
					visited[j] = true;
				}
			}
			sum += num*(num-1)/2;
		}
		return sum;
    }
}

学习别人的代码:

方法一:根据题目给出的数据范围把有序整数拼成一个二位数

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/solution/deng-jie-duo-mi-nuo-gu-pai-dui-de-shu-li-yjlz/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

思路:

       本题中我们需要统计所有等价的多米诺骨牌,其中多米诺骨牌使用二元对代表,「等价」的定义是,在允许翻转两个二元对的的情况下,使它们的元素一一对应相等。于是我们不妨直接让每一个二元对都变为指定的格式,即第一维必须不大于第二维。这样两个二元对「等价」当且仅当两个二元对完全相同。

       注意到二元对中的元素均不大于 99,因此我们可以将每一个二元对拼接成一个两位的正整数,即 (x, y) \to 10x + y(x,y)→10x+y。这样就无需使用哈希表统计元素数量,而直接使用长度为 100100 的数组即可。

class Solution {
    public int numEquivDominoPairs(int[][] dominoes) {
        int[] num = new int[100];
        int ret = 0;
        for (int[] domino : dominoes) {
            int val = domino[0] < domino[1] ? domino[0] * 10 + domino[1] : domino[1] * 10 + domino[0];
            ret += num[val];
            num[val]++;
        }
        return ret;
    }
}

作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/solution/deng-jie-duo-mi-nuo-gu-pai-dui-de-shu-li-yjlz/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

方法二:哈希表

作者:LeetCode
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/solution/deng-jie-duo-mi-nuo-gu-pai-dui-de-shu-li-08z8/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

思路:

  • 把有序数对封装成类,每一个多米诺骨牌就对应了类的一个对象;
  • 在遍历的过程中使用哈希表记录出现的数对的数量。注意:有序数对中只要出现的字符对应相等或者交叉对应相等,在哈希表中就认为它们相等,因此须要 重写 hashCode() 方法、 equlas() 方法。例如 [1, 2] 和 [2, 1] 就须要认为是等价的对象;
  • 假设某一类「等价」的对象的总数为 N,这一类中任意取出 2 个的组合数 C_{N}^{2}=\frac{N(N-1)}{2}  就是这一类对总的「满足 dominoes[i] 和 dominoes[j] 等价的骨牌对 (i, j) 的数量」。对每一类的频数依次求频数对 2 的组合数,再求和即可。
public class Solution {

    public int numEquivDominoPairs(int[][] dominoes) {
        // 为了避免哈希表自动扩容,根据题目的数据范围,设置哈希表初始化的大小为 100
        // Pair 类重写了 hashCode() 和 equals() 方法
        Map<Pair, Integer> freq = new HashMap<>(100);
        for (int[] dominoe : dominoes) {
            Pair key = new Pair(dominoe[0], dominoe[1]);
            freq.put(key, freq.getOrDefault(key, 0) + 1);
        }

        // 根据组合数公式 C_n^2 = (n * (n - 1)) / 2 计算等价骨牌能够组成的组合数,再求和
        int count = 0;
        for (int f : freq.values()) {
            count += (f * (f - 1)) / 2;
        }
        return count;
    }

    private class Pair {

        private int key;
        private int value;

        public Pair(int key, int value) {
            this.key = key;
            this.value = value;
        }

        /**
         * 让有序数对 [a, b] 和 [b, a] 认为是相等的对象
         *
         * @param o
         * @return
         */
        @Override
        public boolean equals(Object o) {
            if (this == o) {
                return true;
            }
            if (o == null || getClass() != o.getClass()) {
                return false;
            }
            Pair pair = (Pair) o;
            return key == pair.key && value == pair.value || key == pair.value && value == pair.key;
        }

        /**
         * 让相同的数对映射到同一个位置
         *
         * @return
         */
        @Override
        public int hashCode() {
            if (key > value) {
                return value * 10 + key;
            }
            return key * 10 + value;
        }
    }
}

作者:LeetCode
链接:https://leetcode-cn.com/problems/number-of-equivalent-domino-pairs/solution/deng-jie-duo-mi-nuo-gu-pai-dui-de-shu-li-08z8/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值