在此程序中,您將學(xué)習(xí)使用Kotlin中的函數(shù)將兩個(gè)矩陣相乘。
為了進(jìn)行矩陣乘法,第一矩陣的列數(shù)必須等于第二矩陣的行數(shù)。在我們的示例中,即
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)) //兩個(gè)矩陣相乘 val product = multiplyMatrices(firstMatrix, secondMatrix, r1, c1, c2) // 顯示結(jié)果 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("兩個(gè)矩陣的乘積是: ") for (row in product) { for (column in row) { print("$column ") } println() } }
運(yùn)行該程序時(shí),輸出為:
兩個(gè)矩陣的乘積是: 24 29 6 25
在上面的程序中,有兩個(gè)函數(shù):
multiplyMatrices() - 將兩個(gè)給定矩陣相乘并返回乘積矩陣
displayProduct() - 在屏幕上顯示乘積矩陣的輸出。
乘法發(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代碼:使用函數(shù)將兩個(gè)矩陣相乘的Java程序