在此程序中,您将学习使用Kotlin中的函数将两个矩阵相乘。
为了进行矩阵乘法,第一矩阵的列数必须等于第二矩阵的行数。在我们的示例中,即
c1 = r2
同样,最终乘积矩阵的大小为r1 x c2,即
product[r1][c2]
您也可以将两个不带函数的矩阵相乘。
fun main(args: Array<String>) { val r1 = 2 val c1 = 3 val r2 = 3 val c2 = 2 val firstMatrix = arrayOf(intArrayOf(3, -2, 5), intArrayOf(3, 0, 4)) val secondMatrix = arrayOf(intArrayOf(2, 3), intArrayOf(-9, 0), intArrayOf(0, 4)) //两个矩阵相乘 val product = multiplyMatrices(firstMatrix, secondMatrix, r1, c1, c2) // 显示结果 displayProduct(product) } fun multiplyMatrices(firstMatrix: Array, secondMatrix: Array, r1: Int, c1: Int, c2: Int): Array{ val product = Array(r1) { IntArray(c2) } for (i in 0..r1 - 1) { for (j in 0..c2 - 1) { for (k in 0..c1 - 1) { product[i][j] += firstMatrix[i][k] * secondMatrix[k][j] } } } return product } fun displayProduct(product: Array) { println("两个矩阵的乘积是: ") for (row in product) { for (column in row) { print("$column ") } println() } }
运行该程序时,输出为:
两个矩阵的乘积是: 24 29 6 25
在上面的程序中,有两个函数:
multiplyMatrices() - 将两个给定矩阵相乘并返回乘积矩阵
displayProduct() - 在屏幕上显示乘积矩阵的输出。
乘法发生为:
|- (a11 x b11) + (a12 x b21) + (a13 x b31) (a11 x b12) + (a12 x b22) + (a13 x b32) -| |_ (a21 x b11) + (a22 x b21) + (a23 x b31) (a21 x b12) + (a22 x b22) + (a23 x b32) _|
在我们的示例中,它发生为:
|- (3 x 2) + (-2 x -9) + (5 x 0) = 24 (3 x 3) + (-2 x 0) + (5 x 4) = 29 -| |_ (3 x 2) + ( 0 x -9) + (4 x 0) = 6 (3 x 3) + ( 0 x 0) + (4 x 4) = 25 _|
这是等效的Java代码:使用函数将两个矩阵相乘的Java程序