1572. 矩阵对角线元素的和(C++)
- 1 题目描述
- 2 示例描述
- 2.1 示例 1
- 2.2 示例 2
- 2.3 示例 3
- 3 解题提示
- 4 源码详解(C++)
1 题目描述
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
2 示例描述
2.1 示例 1
2.2 示例 2
输入:mat =
[[1,1,1,1],
[1,1,1,1],
[1,1,1,1],
[1,1,1,1]]
输出:8
2.3 示例 3
输入:mat = [[5]]
输出:5
3 解题提示
n == mat.length == mat[i].length
1 <= n <= 100
1 <= mat[i][j] <= 100
4 源码详解(C++)
class Solution {
public:
int diagonalSum(vector<vector<int>>& mat) {
int res = 0 ;
if ( mat.size() % 2 == 0)
{
for (int i = 0 ; i < mat.size() ; i ++)
{
for (int j = 0 ; j < mat.size() ; j ++)
{
if ( i == j || i + j == (mat.size() - 1))
{
res = res + mat[i][j];
}
}
}
}
else
{
for (int i = 0 ; i < mat.size() ; i ++)
{
for (int j = 0 ; j < mat.size() ; j ++)
{
if ( i == j || i + j == (mat.size() - 1))
{
res = res + mat[i][j];
}
}
}
}
return res;
}
};