1128. 等价多米诺骨牌对的数量
题目:
给你一组多米诺骨牌 dominoes
。
形式上,dominoes[i] = [a, b]
与 dominoes[j] = [c, d]
等价 当且仅当 (a == c
且 b == d
) 或者 (a == d
且 b == c
) 。即一张骨牌可以通过旋转 0
度或 180
度得到另一张多米诺骨牌。
在 0 <= i < j < dominoes.length
的前提下,找出满足 dominoes[i]
和 dominoes[j]
等价的骨牌对 (i, j)
的数量。
示例 1:
输入:dominoes = [[1,2],[2,1],[3,4],[5,6]]
输出:1
示例 2:
输入:dominoes = [[1,2],[1,2],[1,1],[1,2],[2,2]]
输出:3
提示:
1 <= dominoes.length <= 4 * 104
dominoes[i].length == 2
1 <= dominoes[i][j] <= 9
就这么想:【1,2】和【2,1】是相等的,也就是说。我们可以将每个都变成先小后大的样子,存到一个哈希表中。如果提供的顺序是【1,2】,【1,2】,【1,2】。那么满足题意的有几个呢?,就是第二次及以后出现该数组对时,才计算结果,第一次出现就是0,第二次出现就是1,所以结果是,0 +1 +2总共是三次,所以代码可以如下:
C#:
public class Solution {
public int NumEquivDominoPairs(int[][] dominoes) {
int ans = 0;
int[][] cnt = new int[10][];
for(int i = 0; i < cnt.GetLength(0); i++) {
cnt[i] = new int[10];
}
foreach(var d in dominoes) {
int a = Math.Min(d[0], d[1]);
int b = Math.Max(d[0], d[1]);
ans += cnt[a][b]++;
}
return ans;
}
}
C++:
class Solution {
public:
int numEquivDominoPairs(vector<vector<int>>& dominoes) {
int ans = 0;
int cnt[10][10]{};
for (auto& d : dominoes) {
auto [a, b] = minmax(d[0], d[1]); // 保证 a <= b
ans += cnt[a][b]++;
}
return ans;
}
};