LeetCode2352相等行列对

题目描述

  给你一个下标从 0 开始、大小为 n x n 的整数矩阵 grid ，返回满足 Ri 行和 Cj 列相等的行列对 (Ri, Cj) 的数目。如果行和列以相同的顺序包含相同的元素（即相等的数组），则认为二者是相等的。

解析

  针对题目给出的数量级，暴力是最合适的，过于简单就不详述了，如果数量级过大，可以将每行每列的元素合成字符串然后通过set比较

public int equalPairs(int[][] grid) {
        int res = 0;
        Map<String, Integer> rowMap = new HashMap<>();

        // 将每一行的数据转换为字符串并存储在哈希表中，记录每种模式的出现次数
        for (int[] row : grid) {
            StringBuilder sb = new StringBuilder();
            for (int value : row) {
                sb.append(value).append(",");  // 添加分隔符防止数字合并
            }
            String rowString = sb.toString();
            rowMap.put(rowString, rowMap.getOrDefault(rowString, 0) + 1);
        }

        // 遍历每一列，将列数据转换为与行相同的字符串格式
        for (int j = 0; j < grid[0].length; j++) {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < grid.length; i++) {
                sb.append(grid[i][j]).append(",");
            }
            String colString = sb.toString();
            // 如果行哈希表中存在该列字符串，则增加相应的配对数量
            res += rowMap.getOrDefault(colString, 0);
        }

        return res;
    }

  这种方式在题目的输入下其实可以更加简化，不转为字符串而是转为数字，然后比较数字是否相等，可以大大减少复杂度。
public int equalPairs(int[][] grid) {
int n = grid.length;
int[] row = new int[n];
int[] col = new int[n];
for(int i=0;i<n;i++){
row[i] = grid[i][0];
col[i] = grid[0][i];
for(int j=1;j<n;j++){
row[i] += grid[i][j] * j * 10;
col[i] += grid[j][i] * j * 10;
}
}
int num = 0;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++){
if(row[i] == col[j]){
num++;
}
}
}

    return num;   
}

  此题还有前缀树解法，，通过树形结构记录每行的元素，然后对每列在树上查找，对于大量数据效率较高，此题时间复杂度高。