华为机试-矩阵乘法

Posted 2020-09-25

题目描述
如果A是个x行y列的矩阵，B是个y行z列的矩阵，把A和B相乘，其结果将是另一个x行z列的矩阵C。
这个矩阵的每个元素是由下面的公式决定的：

原型：
voidmatrix_multiply(int *m1,int *m2,int *r, int x, int y, int z);
输入参数：
int *m1：x行y列的矩阵(array1[x][y])
int *m2：y行z列的矩阵(array2[y][z])
int x：矩阵m1的行数
int y：矩阵m1的列数/矩阵m2的行数
int z：矩阵m2的列数

输出参数：
int *r：矩阵m1, m2相乘的结果(array3[x][z])

返回值：
void
输入描述:
输入说明：
1、第一个矩阵的行数x
2、第一个矩阵的列数和第二个矩阵的行数y
3、第二个矩阵的列数z
4、第一个矩阵的值
5、第二个矩阵的值

输出描述:
输出两个矩阵相乘的结果

输入例子:
2
2
2
3 8
8 0
9 0
18 9

输出例子:
171 72
72 0

Java程序实现

 

1. import java.util.Scanner;  
2.   
3. public class Main {  
4.     public static void main(String[] args) {  
5.         Scanner sc = new Scanner(System.in);  
6.   
7.         while (sc.hasNext()) {  
8.             int x = sc.nextInt();  
9.             int y = sc.nextInt();  
10.             int z = sc.nextInt();  
11.             int[][] array1 = new int[x][y];  
12.             int[][] array2 = new int[y][z];  
13.             for (int i = 0; i < x; i++) {  
14.                 for (int j = 0; j < y; j++) {  
15.                     array1[i][j] = sc.nextInt();  
16.                 }  
17.             }  
18.   
19.             for (int i = 0; i < y; i++) {  
20.                 for (int j = 0; j < z; j++) {  
21.                     array2[i][j] = sc.nextInt();  
22.                 }  
23.             }  
24.   
25.             int[][] array3 = new int[x][z];  
26.             matrix_multiply(array1, array2, array3);  
27.   
28.             for (int i = 0; i < x; i++) {  
29.                 for (int j = 0; j < z; j++) {  
30.                     if (j == z - 1) {  
31.                         System.out.println(array3[i][j]);  
32.                     } else {  
33.                         System.out.print(array3[i][j] + " ");  
34.                     }  
35.   
36.                 }  
37.             }  
38.   
39.         }  
40.   
41.     }  
42.   
43.     private static void matrix_multiply(int[][] array1, int[][] array2, int[][] array3) {  
44.         int x = array1.length;  
45.         int y = array2.length;  
46.         int z = array3[0].length;  
47.         // System.out.println(x + " " + y + " " + z);  
48.         for (int i = 0; i < x; i++) {  
49.             for (int j = 0; j < z; j++) {  
50.                 for (int j2 = 0; j2 < y; j2++) {  
51.   
52.                     array3[i][j] += array1[i][j2] * array2[j2][j];  
53.                 }  
54.             }  
55.         }  
56.   
57.     }  
58.   
59. }