題目描述
??給你一個下標從 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;
}
??此題還有前綴樹解法,,通過樹形結構記錄每行的元素,然后對每列在樹上查找,對于大量數據效率較高,此題時間復雜度高。