找出相同的行和列有几对。
相同指的是一整行和一整列的元素都相同。
思路:
把一整行一整列的元素以String的形式保存进HashMap.
然后找是否存在这个String即可。
public int equalPairs(int[][] grid) {
int n = grid.length;
int res = 0;
HashMap<String, Integer> map = new HashMap<>();
for(int i = 0; i < n; i++) {
StringBuilder sb = new StringBuilder();
for(int j = 0; j < n; j++) {
sb.append(grid[i][j]);
sb.append(",");
}
map.put(sb.toString(), map.getOrDefault(sb.toString(), 0)+1);
}
for(int i = 0; i < n; i++) {
StringBuilder sb = new StringBuilder();
for(int j = 0; j < n; j++) {
sb.append(grid[j][i]);
sb.append(",");
}
//System.out.println(sb.toString());
res += map.getOrDefault(sb.toString(), 0);
}
return res;
}
用char数组比StringBuilder更快。
但是很好奇题目中说grid[i][j] <= 105, 怎么没有超出char的范围。
public int equalPairs(int[][] grid) {
int res = 0;
Map<String, Integer> mp = new HashMap<>();
for (int col = 0; col < grid[0].length; col++) {
char[] colArr = new char[grid.length];
for (int row = 0; row < grid.length; row++) {
colArr[row] = (char) grid[row][col];
}
String colStr = String.valueOf(colArr);
mp.put(colStr, mp.getOrDefault(colStr, 0) + 1);
}
for (int row = 0; row < grid.length; row++) {
char[] rowArr = new char[grid[0].length];
for (int col = 0; col < grid[0].length; col++) {
rowArr[col] = (char) grid[row][col];
}
String rowStr = String.valueOf(rowArr);
if (mp.containsKey(rowStr)) {
res += mp.get(rowStr);
}
}
return res;
}