flag
蓝桥杯第17天
题目介绍
问题描述
输入两个矩阵,分别是ms,sn大小。输出两个矩阵相乘的结果。
输入格式
第一行,空格隔开的三个正整数m,s,n(均不超过200)。
接下来m行,每行s个空格隔开的整数,表示矩阵A(i,j)。
接下来s行,每行n个空格隔开的整数,表示矩阵B(i,j)。
输出格式
m行,每行n个空格隔开的整数,输出相乘後的矩阵C(i,j)的值。
样例输入
2 3 2
1 0 -1
1 1 -3
0 3
1 2
3 1
样例输出
-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
思路
读取输入数据到两个二维数组中,再根据乘法公式进行计算即可
一次就通过(1.0s擦边,毕竟三重循环)
代码
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner input=new Scanner(System.in);
int m=input.nextInt();
int s=input.nextInt();
int n=input.nextInt();
long [][]x1=new long[m][s];
long [][]x2=new long[s][n];
for(int i=0;i<m;i++){//将第一个矩阵的值读取到二维数组x1
for(int j=0;j<s;j++){
x1[i][j]=input.nextLong();
}
}
for(int i=0;i<s;i++){//将第二个矩阵的值读取到二维数组x2
for(int j=0;j<n;j++){
x2[i][j]=input.nextLong();
}
}
long [][]res=new long[m][n];//保存乘积的二维数组
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
for(int k=0;k<s;k++){
res[i][j]+=x1[i][k]*x2[k][j];
}
}
}
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
System.out.print(res[i][j]+" ");
}
System.out.println();
}
}
}
tip
当数据量变大时可能超时,这是因为Scanner在数据量大的时候效率低
这是java语言运行超时的一些解决办法:
https://blog.csdn.net/richenyunqi/article/details/84350768
import java.io.*;
public class Main2 {
public static void main(String[] args) throws IOException {
BufferedReader input = new BufferedReader(new InputStreamReader(System.in));
String[] s = input.readLine().split(" ");//将读取的一行字符串按空格分割成几个字符串
int M = Integer.parseInt(s[0]), S = Integer.parseInt(s[1]), N = Integer.parseInt(s[2]);//将字符串转换为整数
int[][] A1 = new int[M][S];
int[][] A2 = new int[S][N];
int[][] A3 = new int[M][N];
for (int i = 0; i < M; ++i) {
s = input.readLine().split(" ");//将读取的一行字符串按空格分割成几个字符串
for (int j = 0; j < S; ++j)
A1[i][j] = Integer.parseInt(s[j]);//将字符串转换为整数
}
for (int i = 0; i < S; ++i){
s = input.readLine().split(" ");//将读取的一行字符串按空格分割成几个字符串
for (int j = 0; j < N; ++j)
A2[i][j] = Integer.parseInt(s[j]);//将字符串转换为整数
}
for (int i = 0; i < M; ++i)
for (int j = 0; j < N; ++j) {
A3[i][j] = 0;
for (int k = 0; k < S; ++k)
A3[i][j] += A1[i][k] * A2[k][j];
}
for (int i = 0; i < M; ++i) {
for (int j = 0; j < N; ++j)
System.out.print(+A3[i][j] + " ");
System.out.println();
}
}
}