华为OJ——矩阵乘法

题目描述

　　如果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、第一个矩阵的行数

2、第一个矩阵的列数和第二个矩阵的行数

3、第二个矩阵的列数

4、第一个矩阵的值

5、第二个矩阵的值

输出描述:

输出两个矩阵相乘的结果

示例1

输入

　　2

　　2

　　2

　　3 8

　　8 0

　　9 0

　　18 9

　　

输出

　　171 72

　　72 0

实现代码

思路：

① 第一个矩阵行元素*第二个矩阵列元素，得到结果矩阵的元素

② 注意结果矩阵，循环的关系

package cn.c_shuang.test6;

import java.util.Scanner;
/**
* 矩阵乘法
* @author Cshuang
*
*/
public class Main {
public static void main(String[] args){
Scanner in = new Scanner(System.in);
while(in.hasNext()){
int x1=in.nextInt();
int x2=in.nextInt();
int x3=in.nextInt();
int[][] a=new int[x1][x2];
int[][] b=new int[x2][x3];
for (int i = 0; i < x1; i++) {
for (int j = 0; j < x2; j++) {
a[i][j]=in.nextInt();
}
}
for (int i = 0; i < x2; i++) {
for (int j = 0; j < x3; j++) {
b[i][j]=in.nextInt();
}
}
caculate(a,b,x1,x2,x3);

}
in.close();
}

private static void caculate(int[][] a, int[][] b, int x1, int x2, int x3) {
int[][]c=new int[x1][x3];
for (int i = 0; i < x1; i++) {
StringBuilder sb=new StringBuilder();
for (int j = 0; j < x3; j++) {
for (int k = 0; k < x2; k++) {
c[i][j]+=a[i][k]*b[k][j];
}
sb.append(c[i][j]+" ");
}
System.out.println(sb.toString().trim());
}
}
}