Python计算矩阵主对角线元素之和：简洁高效的解决方案

以下是使用 Python 计算矩阵主对角线元素之和的代码：pythonclass Solution: def diagonalSum(self, mat: List[List[int]]) -> int: ans = 0 n = len(mat) for i, row in enumerate(mat): j = n - i - 1 ans += row[i] + (0 if j == i else row[j]) return ans

代码解析：

定义类和方法: 代码定义了一个名为 Solution 的类，其中包含一个名为 diagonalSum 的方法。该方法用于计算矩阵主对角线元素之和。2. 参数和初始化: diagonalSum 方法接受一个名为 mat 的二维列表作为参数，表示输入的矩阵。变量 ans 初始化为 0，用于存储主对角线元素之和。3. 获取矩阵大小: n = len(mat) 获取矩阵的行数，也代表矩阵的列数（假设矩阵是方阵）。4. 遍历矩阵: 代码使用 for i, row in enumerate(mat) 遍历矩阵的每一行，并使用 enumerate 函数获取行索引 i 和对应的行 row。5. 计算对角线元素索引: j = n - i - 1 计算当前行中需要加到 ans 中的元素的列索引 j。由于主对角线上的元素的行索引和列索引相等或对称，所以可以通过此公式计算得到 j。6. 累加元素: - row[i] 表示当前行中主对角线上的元素。 - (0 if j == i else row[j]) 用于处理矩阵大小为奇数的情况。当 j == i 时，表示当前元素已经在上一步被加过，因此不需要再次加，所以加 0；否则，将 row[j]（对角线另一端的元素）加到 ans 中。7. 返回结果: 循环结束后，ans 中存储了矩阵主对角线元素之和，方法返回 ans。

时间复杂度:

该代码的时间复杂度为 O(n)，其中 n 为方阵的行数。这是因为代码只需要遍历一次矩阵的所有行即可计算出主对角线元素之和。

总结:

这段代码提供了一种简洁高效的方法来计算矩阵主对角线元素之和。它逻辑清晰，易于理解，并且具有良好的时间复杂度。