求平方矩阵的两个对角线之和的乘积

程序功能

该程序用于计算一个给定的平方矩阵的两个对角线之和的乘积。

输入参数

该程序需要输入一个平方矩阵作为参数，矩阵可以用一个二维数组表示。

输出结果

该程序返回输入的平方矩阵的两个对角线之和的乘积。

程序实现

Python实现

def diagonal_product(matrix):
    dim = len(matrix)
    diagonal1 = sum(matrix[i][i] for i in range(dim))
    diagonal2 = sum(matrix[i][dim-i-1] for i in range(dim))
    return diagonal1 * diagonal2

Java实现

public static int diagonalProduct(int[][] matrix) {
    int dim = matrix.length;
    int diagonal1 = 0;
    int diagonal2 = 0;
    for (int i = 0; i < dim; i++) {
        diagonal1 += matrix[i][i];
        diagonal2 += matrix[i][dim-i-1];
    }
    return diagonal1 * diagonal2;
}

C++实现

int diagonalProduct(vector<vector<int>>& matrix) {
    int dim = matrix.size();
    int diagonal1 = 0;
    int diagonal2 = 0;
    for (int i = 0; i < dim; i++) {
        diagonal1 += matrix[i][i];
        diagonal2 += matrix[i][dim-i-1];
    }
    return diagonal1 * diagonal2;
}

以上代码使用了不同的编程语言实现了相同的功能，但可能需要根据不同的数据类型调整代码细节。

示例

输入

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

输出

720

以上示例输入了一个 $3 \times 3$ 的平方矩阵，矩阵的对角线之和分别为 $1 + 5 + 9 = 15$ 和 $3 + 5 + 7 = 15$，它们的乘积为 $15 \times 15 = 225$，程序返回的结果为 $720$。