题目:1128. 等价多米诺骨牌对的数量
思路:哈希表,时间复杂度0(n)。
每个元素值到小于10,那么可以直接记录x*10+y出现的情况。因为可以旋转,所以可以维持x<=y即可。
C++版本:
class Solution {
public:
int numEquivDominoPairs(vector<vector<int>>& dominoes) {
unordered_map<int,int> mp;
int ans=0;
for(int i=0;i<dominoes.size();i++){
int x=dominoes[i][0],y=dominoes[i][1];
if(x>y) swap(x,y);
ans+=mp[x*10+y];
mp[x*10+y]++;
}
return ans;
}
};
JAVA版本:
class Solution {
public int numEquivDominoPairs(int[][] dominoes) {
Map<Integer,Integer> mp= new HashMap<Integer,Integer>();
int ans=0;
for(int i=0;i<dominoes.length;i++){
int x=dominoes[i][0],y=dominoes[i][1];
if(x>y){
int t=x;
x=y;
y=t;
}
ans+=mp.getOrDefault(x*10+y,0);
mp.put(x*10+y,mp.getOrDefault(x*10+y,0)+1);
}
return ans;
}
}
Go版本:
func numEquivDominoPairs(dominoes [][]int) int {
mp:=map[int]int{}
ans:=0
for i:= range dominoes {
x,y:=dominoes[i][0],dominoes[i][1]
if x>y {
t:=x
x=y
y=t
}
ans+=mp[x*10+y]
mp[x*10+y]++
}
return ans
}