如何计算NxN矩阵的主对角线元素和反向对角线元素之和？

计算NxN矩阵的主对角线元素和反向对角线元素之和，可以使用以下代码实现：

# 计算NxN矩阵的主对角线元素和反向对角线元素之和
def diagonal_sum(matrix):
    n = len(matrix)
    sum = 0
    for i in range(n):
        sum += matrix[i][i]  # 主对角线元素之和
        sum += matrix[i][n-i-1]  # 反向对角线元素之和
    return sum

其中，`matrix`表示输入的NxN矩阵，`n`表示矩阵的大小，`sum`表示主对角线元素和反向对角线元素之和。