矩阵对角线元素的和

题目:

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

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

示例:

输入:mat = \[1,2,3,

4,5,6,

7,8,9\]

输出:25

解释:对角线的和为:1 + 5 + 9 + 3 + 7 = 25

请注意,元素 mat11 = 5 只会被计算一次。

来源:力扣(LeetCode)

链接:https://leetcode.cn/problems/matrix-diagonal-sum

python 复制代码
class Solution:
    def diagonalSum(self, mat):
        j=0
        c=1
        a=[]
        b=len(mat)
        for i in mat:
            a.append(i[j])
            a.append(i[len(mat[0])-c])
            c+=1
            j+=1
        d=sum(a)
        if b%2 == 1:
            d-= mat[b/2][b/2]
        return d
相关推荐
Warson_L9 小时前
Python `Annotated` 与 LangGraph Reducer 学习笔记
python
韩师傅9 小时前
海天线算法的前世今生
python·计算机视觉
韩师傅9 小时前
当你的甲方设备过烂,要如何快速出效果?
python·计算机视觉
Warson_L9 小时前
LangGraph的MessageState and HumanMessage
python
韩师傅10 小时前
当你的甲方吐槽天空不够蓝,你应该如何应对
python·计算机视觉
Warson_L10 小时前
python的类&继承
python
Warson_L11 小时前
类型标注/type annotation
python
ThreeS13 小时前
手搓MiniVLA全实战教程-一步一步用pytorch解释原理与思路
人工智能·python
金銀銅鐵14 小时前
[Python] 模 n 乘法的逆元计算器
python·数学·游戏