在此程序中,您將學(xué)習(xí)在Kotlin中使用多維數(shù)組將兩個矩陣相乘。
為了進行矩陣乘法,第一矩陣的列數(shù)必須等于第二矩陣的行數(shù)。在我們的示例中,即
c1 = r2
因此,最終矩陣乘積的大小為r1 x c2,即
product[r1][c2]
您也可以使用函數(shù)將兩個矩陣相乘。
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 = 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] } } } //顯示結(jié)果 println("兩個矩陣的乘積是: ") for (row in product) { for (column in row) { print("$column ") } println() } }
運行該程序時,輸出為:
兩個矩陣的乘積是: 24 29 6 25
在上面的程序中,乘法發(fā)生為:
|- (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) _|
在我們的示例中,它發(fā)生為:
|- (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程序