给你一个下标从 0 开始、大小为 n x n
的整数矩阵 grid
,返回满足 Ri
行和 Cj
列相等的行列对 (Ri, Cj)
的数目。
如果行和列以相同的顺序包含相同的元素(即相等的数组),则认为二者是相等的。
示例 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].length
1 <= n <= 200
1 <= grid[i][j] <= 105
class Solution {
public int equalPairs(int[][] grid) {
// 存储行
Map<List<Integer>,Integer> rowList=new HashMap<>();
for(int[] row:grid){
// 存储每一行的数据
List<Integer> rows=new ArrayList<>();
for(int i:row){
rows.add(i);
}
// 将行存入map中,若有多行键+1
rowList.put(rows,rowList.getOrDefault(rows,0)+1);
}
int count=0;
for(int i=0;i<grid.length;i++){
// 存储列
List<Integer> cols=new ArrayList<>();
for(int j=0;j<grid.length;j++){
cols.add(grid[j][i]);
}
count+=rowList.getOrDefault(cols,0);
}
return count;
}
}