Python中的Leetcode矩阵对角线和

矩阵对角线和问题是计算矩阵中主对角线和副对角线上所有元素的和。在Leetcode中，这是一个非常常见的问题，因此在Python中解决矩阵对角线和问题也非常重要。

解决方法

有两种常见的解决方法：使用for循环和使用NumPy。

使用for循环

使用for循环的解决方法非常简单明了。我们将矩阵的主对角线和副对角线的元素分别求和。以下是一段使用for循环的代码：

def diagonalSum(mat):
    total_sum = 0
    
    n = len(mat)
    for i in range(n):
        total_sum += mat[i][i] + mat[i][n-1-i]
        
    if n % 2 != 0:
        total_sum -= mat[n//2][n//2]
        
    return total_sum

其中，'mat'是输入的矩阵，'total_sum'是矩阵对角线和。

这个方法的时间复杂度为O(N^2)。

使用NumPy

使用NumPy的解决方法相对简单和高效。以下是一段使用NumPy的代码：

import numpy as np

def diagonalSum(mat):
    diagonal_sum = np.trace(mat) + np.trace(np.fliplr(mat))
    n = len(mat)
    
    if n % 2 != 0:
        diagonal_sum -= mat[n//2][n//2]
        
    return diagonal_sum

其中，'np.trace(mat)'是矩阵的主对角线的和，'np.trace(np.fliplr(mat))'是矩阵的副对角线的和。

这个方法的时间复杂度为O(N)。

总结

无论你使用哪种方法，Python中解决矩阵对角线和问题都非常简单。使用for循环最好适用于小型矩阵，而使用NumPy适用于大型矩阵。无论哪种方法，你都可以在Leetcode等平台中轻松通过此问题。