首页 > 代码库 > 华为机试-矩阵乘法

华为机试-矩阵乘法

题目描述
如果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. }  

 

华为机试-矩阵乘法