试题 算法提高 矩阵相乘
资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
小明最近在为线性代数而头疼,线性代数确实很抽象(也很无聊),可惜他的老师正在讲这矩阵乘法这一段内容。
当然,小明上课打瞌睡也没问题,但线性代数的习题可是很可怕的。
小明希望你来帮他完成这个任务。
现在给你一个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
代码:(日常生活中的大多数矩阵,都够用)
package matrix;
import java.util.Scanner;
public class Main {
public static Scanner sc = new Scanner(System.in);
public static long [][] ma1 = new long [110][110];
public static long [][] ma2 = new long [110][110];
public static long [][] res = new long [110][110];
/**
* 处理一行和一列的乘法运算
*/
public static void f(int row, int col, long len) {
long sum = 0;
for(int i = 1;i <= len;i++) {
sum += (ma1[row][i] * ma2[i][col]);
}
res[row][col] = sum;
}
/**
* 矩阵乘法
*/
public static void multi(long m, long n, long m2, long n2) {
for(int i = 1;i <= m;i++) {
for(int j = 1;j <= n2;j++) {
f(i, j, n);
}
}
}
/**
* 打印矩阵res
* @param args
*/
public static void show(long m, long n2) {
for(int i = 1;i <= m;i++) {
for(int j = 1;j <= n2;j++) {
System.out.print(res[i][j] + " ");
}
System.out.println();
}
}
public static void main(String[] args) {
long m, n;
m = sc.nextLong();n = sc.nextLong();
for(int i = 1;i <= m;i++) {
for(int j = 1;j <= n;j++) {
ma1[i][j] = sc.nextLong();
}
}
long m2, n2;
m2 = sc.nextLong();n2 = sc.nextLong();
for(int i = 1;i <= m2;i++) {
for(int j = 1;j <= n2;j++) {
ma2[i][j] = sc.nextLong();
}
}
multi(m, n, m2, n2);
show(m, n2);
}
}