算法提高 矩阵相乘

来源:互联网 发布:vivo手机一键锁屏软件 编辑:程序博客网 时间:2024/06/08 20:01

问题描述
  小明最近在为线性代数而头疼,线性代数确实很抽象(也很无聊),可惜他的老师正在讲这矩阵乘法这一段内容。
  当然,小明上课打瞌睡也没问题,但线性代数的习题可是很可怕的。
  小明希望你来帮他完成这个任务。

  现在给你一个ai行aj列的矩阵和一个bi行bj列的矩阵,
  要你求出他们相乘的积(当然也是矩阵)。
  (输入数据保证aj=bi,不需要判断)
输入格式
  输入文件共有ai+bi+2行,并且输入的所有数为整数(long long范围内)。
  第1行:ai 和 aj
  第2~ai+2行:矩阵a的所有元素
  第ai+3行:bi 和 bj
  第ai+3~ai+bi+3行:矩阵b的所有元素
输出格式
  输出矩阵a和矩阵b的积(矩阵c)
  (ai行bj列)
样例输入
2 2
12 23
45 56
2 2
78 89
45 56
样例输出
1971 2356
6030 7141

import java.util.Scanner;public class Main{    public static void main(String args[]){        Scanner in=new Scanner(System.in);        int m1=in.nextInt();        int n1=in.nextInt();        int a[][]=new int[m1][n1];        for(int i=0;i<m1;i++){            for(int j=0;j<n1;j++){                a[i][j]=in.nextInt();            }        }        int m2=in.nextInt();        int n2=in.nextInt();        int b[][]=new int[m2][n2];        for(int i=0;i<m2;i++){            for(int j=0;j<n2;j++){                b[i][j]=in.nextInt();            }        }        matrixMul(a,b);    }    public static void matrixMul(int a[][],int b[][]){        int m=a.length;        int n=b[0].length;        int res[][]=new int[m][n];        int sum=0;              for(int i=0;i<m;i++){            for(int k=0;k<n;k++){                 for(int j=0;j<b.length;j++){                    sum+=a[i][j]*b[j][k];                }                res[i][k]=sum;                sum=0;                  }               }           for(int i=0;i<m;i++){            for(int j=0;j<n;j++){                System.out.print(res[i][j]+" ");            }            System.out.println();        }    }}  
0 0