当前位置：首页 > news >正文

LeetCode 1128. 等价多米诺骨牌对的数量题解

news 来源：原创 2025/5/7 10:35:51

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

通过示例观察我们可以发现[1,2]有三个，[1,1]有一个，[2,2]有一个，而等价多米诺骨牌对数量的计算是通过计数的方式统计的，如果是三个那就sum=2+1，ans+=sum即可，如果为四个，那就sum=3+2+1，ans+=sum，而在本题中[1,2]，[2,1]可以看作同一个数，一起统计进去即可，那我这题选择使用map进行维护，当然有数组的话就不能使用传统的Map<Integer,Integer>,可以使用比较好改变的Map<String,Integer>来维护，String的值的话可以使用(i,j)或者(j,i)，反正就是哪个先来用哪个即可。

那我就分段解释一下我都代码逻辑

        int n = dominoes.length;HashMap<String,Integer> hashmap = new HashMap<>();for(int i=0;i<n;i++){String key = "("+ dominoes[i][0] + "," + dominoes[i][1] + ")";String keyR = "(" + dominoes[i][1] + "," + dominoes[i][0] + ")";if(hashmap.getOrDefault(key,0)>0){hashmap.put(key,hashmap.getOrDefault(key,0)+1);}else if(hashmap.getOrDefault(keyR,0)>0){hashmap.put(keyR,hashmap.getOrDefault(keyR,0)+1);}else{hashmap.put(key,hashmap.getOrDefault(key,0)+1);}}

拿[[2,1],[1,2],[1,2]]举一个例子，当[2,1]进入循环，因为一开始都不存在，所以在hashmap中我们会去创建一个新的键值对，key就为(2,1),值为1，当后续遍历至[1,2],因为已经存在[2,1]这个健，我们就会往keyR这个键的值中进行加加。

        int sum = 0;for(Map.Entry<String,Integer> entry:hashmap.entrySet()){// System.out.println("key="+entry.getKey()+" value="+entry.getValue());int mid = entry.getValue()*(entry.getValue()-1)/2;sum+=mid;}

注释的部分是打印出来对步骤进行调试的，对于上面sum的统计其实是可以统计一个公式的，拿3举个例子，sum=3*（3-1）/2,然后统计即可。hashmap.entrySet()这个用法对于map维护数组的作用还是非常大的，可以记一下。

完整代码如下

class Solution {public int numEquivDominoPairs(int[][] dominoes) {int n = dominoes.length;HashMap<String,Integer> hashmap = new HashMap<>();for(int i=0;i<n;i++){String key = "("+ dominoes[i][0] + "," + dominoes[i][1] + ")";String keyR = "(" + dominoes[i][1] + "," + dominoes[i][0] + ")";if(hashmap.getOrDefault(key,0)>0){hashmap.put(key,hashmap.getOrDefault(key,0)+1);}else if(hashmap.getOrDefault(keyR,0)>0){hashmap.put(keyR,hashmap.getOrDefault(keyR,0)+1);}else{hashmap.put(key,hashmap.getOrDefault(key,0)+1);}}int sum = 0;for(Map.Entry<String,Integer> entry:hashmap.entrySet()){// System.out.println("key="+entry.getKey()+" value="+entry.getValue());int mid = entry.getValue()*(entry.getValue()-1)/2;sum+=mid;}return sum;}
}

看了下官方的题解，对于官方的题解它的健和值分别用的是数组下标和数组的值存的，只能说真的非常秒，但也可能是只针对本题而已，毕竟它的范围是1 <= dominoes[i][j] <= 9，所以数组的边界定义在100，然后根据二维数组的下标将[1][2]变为[12],[2][1]变为[12],判断下标i和j的大小即可很好的判断出来。最后一样统计数量即可。

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;}
}