leetcode3128. 直角三角形-medium

1 标题：直角三角形

官方标定难度：中

给你一个二维 boolean 矩阵 grid 。
如果 grid 的 3 个元素的集合中，一个元素与另一个元素在 同一行，并且与第三个元素在 同一列，则该集合是一个 直角三角形。3 个元素 不必 彼此相邻。
请你返回使用 grid 中的 3 个元素可以构建的 直角三角形 数目，且满意 3 个元素值 都 为 1 。
示例 1：

输入：grid = [[0,1,0],[0,1,1],[0,1,0]]
输出：2
解释：
有 2 个值为 1 的直角三角形。留意蓝色的那个 没有 构成直角三角形，由于 3 个元素在同一列。
示例 2：

输入：grid = [[1,0,0,0],[0,1,0,1],[1,0,0,0]]
输出：0
解释：
没有值为 1 的直角三角形。留意蓝色的那个 没有 构成直角三角形。
示例 3：

输入：grid = [[1,0,1],[1,0,0],[1,0,0]]
输出：2
解释：
有两个值为 1 的直角三角形。
提示：

1 <= grid.length <= 1000
1 <= grid.length <= 1000
0 <= grid[j] <= 1
2 solution

代码

1. class Solution {
2. public:
3.     long long numberOfRightTriangles(vector<vector<int>> &grid) {
4.         int m = grid.size(), n = grid[0].size();
5.         vector<int> row(m), col(n);
6.         long long s = 0;
7.         for (int i = 0; i < m; i++) {
8.             for (int j = 0; j < n; j++) {
9.                 row[i] += grid[i][j];
10.                 col[j] += grid[i][j];
11.             }
12.         }
14.         for (int i = 0; i < m; i++) {
15.             for (int j = 0; j < n; j++) {
16.                 if (grid[i][j]) {
17.                     s += (row[i] - 1ll) * (col[j] - 1);   
18.                 }
19.             }
20.         }
21.         return s;
22.     };
23. };

复制代码

结果

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。