2352. 相等行列对 - 力扣（LeetCode）
2023_06_06
[toc]
给你一个下标从 0 开始、大小为 n x n 的整数矩阵 grid ，返回满足 Ri 行和 Cj 列相等的行列对 (Ri, Cj) 的数目。
如果行和列以相同的顺序包含相同的元素（即相等的数组），则认为二者是相等的。
示例 1：
输入：grid = [[3,2,1],[1,7,6],[2,7,7]]
输出：1
解释：存在一对相等行列对：
示例 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
来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/equal-row-and-column-pairs
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
Solution
cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
| class Solution {
public:
int equalPairs(vector<vector<int>>& grid) {
int res = 0, n = grid.size();
for (int row = 0; row < n; row++)
{
for (int col = 0; col < n; col++)
{
if(equal(row, col, grid))
{
res++;
}
}
}
return res;
}
bool equal(int row, int col, vector<vector<int>> &grid)
{
int n = grid.size();
for (int i = 0; i < n; i++)
{
if(grid[row][i] != grid[i][col])
{
return false;
}
}
return true;
}
};
|
Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
| func equalPairs(grid [][]int) int {
n := len(grid)
cnt := make(map[string]int)
for _,row := range grid {
cnt[fmt.Sprint(row)]++
}
res := 0
for j := 0; j < n; j++ {
var arr[] int
for i := 0; i < n; i++ {
arr = append(arr, grid[i][j])
}
if val,ok := cnt[fmt.Sprint(arr)]; ok {
res += val
}
}
return res
}
|
