通过万岁!!!
- 题目:给你一个二维数组,里面只有0或者1。然后让你找到里面的所有的由1构成的直角三角形。值得注意的是,三个1的位置直角的顶点与其他两点在同一行或者同一列。
- 思路:这个题目其实就是确定好直角的那个1就好了,假设这一行有x个元素,这一列有y个元素,那么选定一个以后,还有x-1个,从这x-1个中随便选一个,然后从y-1个中随便选一个。那么这个点所构成的直角三角形的个数是(x-1)*(y-1)。
- 技巧:组合
java代码
java
class Solution {
public long numberOfRightTriangles(int[][] grid) {
Map<Integer, Integer> rowMap = new HashMap<>();// 第key行有多少个1
Map<Integer, Integer> colMap = new HashMap<>();// 第key列有多少个1
for (int i = 0; i < grid.length; i++) {
for (int j = 0; j < grid[0].length; j++) {
if (grid[i][j] == 1) {
Integer rowNum = rowMap.getOrDefault(i, 0);
rowMap.put(i, rowNum + 1);
Integer colNum = colMap.getOrDefault(j, 0);
colMap.put(j, colNum + 1);
}
}
}
long ret = 0;
for (int i = 0; i < grid.length; i++) {
if (rowMap.get(i) == null) {
continue;
}
for (int j = 0; j < grid[0].length; j++) {
if (grid[i][j] == 1) {
ret += (rowMap.get(i) - 1) * (colMap.get(j) - 1);
}
}
}
return ret;
}
}
上面的代码时间和空间都不是很好。主要问题是这里没必要用map
java代码------进阶
java
class Solution {
public long numberOfRightTriangles(int[][] grid) {
int n = grid.length;
int m = grid[0].length;
int[] colNum = new int[m];
int[] rowNum = new int[n];
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
if (grid[i][j] == 1) {
colNum[j]++;
rowNum[i]++;
}
}
}
long ret = 0;
for (int i = 0; i < n; i++) {
if (rowNum[i] == 0) {
continue;
}
for (int j = 0; j < m; j++) {
if (grid[i][j] == 1) {
ret += (rowNum[i] - 1) * (colNum[j] - 1);
}
}
}
return ret;
}
}
- 总结:题目不是特别难,如果这个矩阵特别稀疏的话,我觉得还是map的方式会好一点。