华为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());
        }
    }
}