直角三角形

2024-08-02

题目:

给你一个二维 boolean 矩阵 grid

请你返回使用 grid 中的 3 个元素可以构建的 直角三角形 数目,且满足 3 个元素值 为 1 。

注意:

  • 如果 grid 中 3 个元素满足:一个元素与另一个元素在 同一行,同时与第三个元素在 同一列 ,那么这 3 个元素称为一个 直角三角形 。这 3 个元素互相之间不需要相邻。

示例 1:

     
0 1 0
0 1 1
0 1 0
     
0 1 0
0 1 1
0 1 0

输入:grid = [[0,1,0],[0,1,1],[0,1,0]]

输出:2

解释:

有 2 个直角三角形。

提示:

  • 1 <= grid.length <= 1000
  • 1 <= grid[i].length <= 1000
  • 0 <= grid[i][j] <= 1

思路:

寻找直角三角形的个数,关键就在一个点:如何去重

我们访问二维数组中的每一个点,如果为1,那么就可以成为直角顶点,这样在访问不同 矩阵位置的时候,就是以不同的直角顶点为基本,这样就不用考虑去重问题了

统计二维矩阵每一行,每一列的1的个数,这样当 grid[i][j] 为1时,以它为直角顶点的直角三角形个数就是 (该行 1 的个数 -  1) * (该列 1 的个数 - 1)

代码:

class Solution {
public:
    long long numberOfRightTriangles(vector<vector<int>>& grid) {
        int m = grid.size(), n = grid[0].size();
        vector<int> rowCnt(m, 0);
        vector<int> colCnt(n, 0);
        for (int i = 0; i < m; i++) {
            int cnt = 0;
            for (int j = 0; j < n; j++) {
                if (grid[i][j]) cnt++;
            }
            rowCnt[i] = cnt;
        }
        for (int j = 0; j < n; j++) {
            int cnt = 0;
            for (int i = 0; i < m; i++) {
                if (grid[i][j]) cnt++;
            }
            colCnt[j] = cnt;
        }
        long long ans = 0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (grid[i][j]) {
                    ans += (rowCnt[i] - 1) * (colCnt[j] - 1);
                }
            }
        }
        return ans;
    }
};