Java 算法 矩阵乘法

题目描述

输入两个矩阵，分别是ms，sn大小。输出两个矩阵相乘的结果。

输入
输入描述:
　　第一行，空格隔开的三个正整数m,s,n（均不超过200）。
　　接下来m行，每行s个空格隔开的整数，表示矩阵A（i，j）。
　　接下来s行，每行n个空格隔开的整数，表示矩阵B（i，j）。
输入样例:
2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1

输出

输出描述:
　　m行，每行n个空格隔开的整数，输出相乘後的矩阵C（i，j）的值。
输出样例:
-3 2
-8 2

提示
矩阵C应该是m行n列，其中C(i,j)等于矩阵A第i行行向量与矩阵B第j列列向量的内积。
例如样例中C(1,1)=(1,0,-1)(0,1,3) = 1 * 0 +01+(-1)*3=-3

HINT:时间限制：1.0s 内存限制：512.0MB

解题思路

就按照矩阵的乘法规则进行计算就行了，一个矩阵的行乘以一个矩阵的列之后的和。

代码

import java.util.Scanner;

public class Main {
    static int k=0;
    static int x=0;
    static int y=0;
    public static void main(String[] args) {
        Scanner scanner=new Scanner(System.in);
        int m=scanner.nextInt();
        int s=scanner.nextInt();
        int n=scanner.nextInt();
        int [][]a=new int[m][s];
        int [][]b=new int[s][n];
        int [][]c=new int[m][n];
        for (int i=0;i<m;i++){
            for (int j=0;j<s;j++){
                a[i][j]=scanner.nextInt();
            }
        }
        for (int i=0;i<s;i++){
            for (int j=0;j<n;j++){
                b[i][j]=scanner.nextInt();
            }
        }
        for (int i=0;i<m;i++){
            for (int q=0;q<n;q++){
                for (int j=0;j<s;j++){
                    k=a[i][j]*b[j][y];			//第一个矩阵的行乘以第二个矩阵的列
                    x+=k;						//之后相加得到对应的值
                }
                y++;							//第二个矩阵的列累加
                if (y==n){
                    y=0;
                }
                c[i][q]=x;						//填入对应的数
                x=0;
            }
        }
        for (int i=0;i<m;i++){
            for (int j=0;j<n;j++){
                System.out.print(c[i][j]+" ");//输出
            }
            System.out.println();
        }
    }
}