LeetCode刷题（145）～矩阵对角线元素的和

题目描述

给你一个正方形矩阵 mat，请你返回矩阵对角线元素的和。

请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。

示例 1：

输入：mat = [[1,2,3], [4,5,6], [7,8,9]]
输出：25
解释：对角线的和为：1 + 5 + 9 + 3 + 7 = 25
请注意，元素 mat[1][1] = 5 只会被计算一次。

  
 

  
1
  
2
  
3
  
4
  
5
  
6
 

示例 2：

输入：mat = [[1,1,1,1], [1,1,1,1], [1,1,1,1], [1,1,1,1]]
输出：8

  
 

  
1
  
2
  
3
  
4
  
5
 

示例 3：

输入：mat = [[5]]
输出：5

  
 

  
1
  
2
 

提示：

• n == mat.length == mat[i].length
• 1 <= n <= 100
• 1 <= mat[i][j] <= 100

解答 By 海轰

提交代码

int diagonalSum(vector<vector<int>>& mat) { int ans=0; int n=mat.size(); for(int i=0;i<n;++i) ans+=mat[i][i]; for(int i=0;i<n;++i) ans+=mat[i][n-1-i]; if(n%2==1) ans-=mat[n/2][n/2]; return ans; }

  
 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
 

运行结果

提交代码（优化）

int diagonalSum(vector<vector<int>>& mat) { int ans=0; int n=mat.size(); for(int i=0;i<n;++i) { ans+=mat[i][i]; ans+=mat[i][n-1-i]; } if(n%2==1) ans-=mat[n/2][n/2]; return ans; }

  
 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
 

运行结果

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/matrix-diagonal-sum

文章来源: haihong.blog.csdn.net，作者：海轰Pro，版权归原作者所有，如需转载，请联系作者。

原文链接：haihong.blog.csdn.net/article/details/108539721