2352. 相等行列对
难度:中等 时间:2023/06/06
给你一个下标从 0 开始、大小为 n x n 的整数矩阵 grid ,返回满足 Ri 行和 Cj 列相等的行列对 (Ri, Cj) 的数目 。
如果行和列以相同的顺序包含相同的元素(即相等的数组),则认为二者是相等的。
示例 1:
输入:grid = [[3,2,1],[1,7,6],[2,7,7]]
输出:1
解释:存在一对相等行列对:
- (第 2 行,第 1 列):[2,7,7]
示例 2:
输入:grid = [[3,1,2,2],[1,4,4,5],[2,4,2,2],[2,4,2,2]]
输出:3
解释:存在三对相等行列对:
- (第 0 行,第 0 列):[3,1,2,2]
- (第 2 行, 第 2 列):[2,4,2,2]
- (第 3 行, 第 2 列):[2,4,2,2]
提示:
n == grid.length == grid[i].length1 <= n <= 2001 <= grid[i][j] <= 10^5
解题思路:
1.横向遍历矩阵,将每一行数据以字符串格式作为key存入HashMap,value为key出现的次数; 2.纵向遍历矩阵,拿到每一列数据在map中寻找是否有一样的key与其对应,若存在一样的,则满足的情况应为这个key出现的次数-即value种。
class Solution {
public int equalPairs(int[][] grid) {
int len = grid.length;
int ans = 0;
Map<String,Integer> map = new HashMap<>();
for(int i = 0;i < len;i++){
map.put(Arrays.toString(grid[i]),map.getOrDefault(Arrays.toString(grid[i]),0) + 1);
}
int[] arr = new int[len];
for(int i = 0;i < len;i++){
for(int j = 0;j < len;j++){
arr[j] = grid[j][i];
}
if(map.containsKey(Arrays.toString(arr))){
ans += map.get(Arrays.toString(arr));
}
}
return ans;
}
}