LC-2352. 相等行列对（模拟、哈希表）

2352. 相等行列对

难度中等35

给你一个下标从 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].length
• 1 <= n <= 200
• 1 <= grid[i][j] <= 105

模拟

class Solution {
    public int equalPairs(int[][] grid) {
        int n = grid.length;
        int ans = 0;
        for(int i = 0; i < n; i++){
            for(int k = 0; k < n; k++){
                int idx = 0;
                while(idx < n && grid[i][idx] == grid[idx][k])
                    idx++;
                if(idx == n) ans += 1;
            }
        }
        return ans;
    }
}

时间复杂度：O(n^3)

哈希表

用哈希表统计每行出现的次数，然后遍历列，累加哈希表中列出现的次数。

python

class Solution:
    def equalPairs(self, grid: List[List[int]]) -> int:
        cnt = Counter(tuple(row) for row in grid)
        return sum(cnt[col] for col in zip(*grid))

java

java中不能直接统计每行出现的次数，但是可以将每行转为字符串，然后存入map中

• String rowstr = Arrays.toString(row);

class Solution {
    public int equalPairs(int[][] grid) {
        Map<String, Integer> map = new HashMap<>();
        // 统计行出现的次数
        for(int[] row : grid){
            String rowstr = Arrays.toString(row);
            map.put(rowstr, map.getOrDefault(rowstr, 0) + 1);
        }
        // 遍历列，计算答案
        int ans = 0;
        for(int j = 0; j < grid.length; j++){
            int[] col = new int[grid.length];
            int idx = 0;
            for(int[] row : grid){
                col[idx++] = row[j];
            }
            String colstr = Arrays.toString(col);
            ans += map.getOrDefault(colstr, 0);
        }
        return ans;
    }
}