话不多说,又是一道简单的Leetcode分享给大家,先思路后代码
一、首先介绍下题目
给你一个正方形矩阵 mat,请你返回矩阵对角线元素的和。
请你返回在矩阵主对角线上的元素和副对角线上且不在主对角线上元素的和。
例如:
输入:mat = [[1,2,3],
[4,5,6],
[7,8,9]]
输出:25
解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25
请注意,元素 mat[1][1] = 5 只会被计算一次
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/matrix-diagonal-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、阐述一下解题思路
1.其实一开始看到这,脑子是有点懵的,不过仔细看了下,发现Leetcode也挺仁慈的,间接也透露出了一些思路。看到上面的例子,你会想到这道题肯定不会是通过暴力遍历法来解决,而是有规律的,从1,3,5,7,9你就能感觉到
2.然后就可以想到,左右是个对称的关系,能否从对称这个角度来出发,而且选出的元素,左边是[0,0],[1,1]非常有规律,右变就是[0,mat.shape[1]-0],[1,mat.shape[1]-1]非常有规律,那么解题思路就出来了。
总结:利用矩阵的对称性,从元素索引的对称性出发
三、代码过程
class Solution:
def diagonalSum(self, mat: List[List[int]]) -> int:
list_ = []
sum_ = 0
for i in range(len(mat[0])):
if [0+i, 0+i] not in list_:
list_.append([0+i, 0+i])
if [len(mat[0])-1-i, 0+i] not in list_:
list_.append([len(mat[0])-1-i, 0+i])
for i in list_:
number_1 = i[0]
number_2 = i[1]
sum_ = sum_ + mat[number_1][number_2]
return sum_
四、成绩
成绩一般般,有兴趣的同学可以一起改进改进