算法提高 矩阵相乘

来源:互联网 发布:惠惠海淘助手淘宝货 编辑:程序博客网 时间:2024/05/16 09:10

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

  现在给你一个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.io.BufferedInputStream;import java.math.BigInteger;import java.util.Scanner;public class Main {    public static BigInteger[][] Mul(BigInteger[][] A ,int col1,BigInteger[][] B,int col2){        BigInteger[][] C = new BigInteger[A.length][col2];        for ( int i = 0 ; i < A.length ; i++){            for ( int k = 0 ; k < col2 ; k++){                BigInteger sum = BigInteger.ZERO;                for ( int j = 0 ; j < col1 ; j++){                    sum = sum.add(A[i][j].multiply(B[j][k]));                }                C[i][k] = sum;             }        }        return C;    }    public static void main(String[] args) {        // TODO Auto-generated method stub        Scanner in = new Scanner(new BufferedInputStream(System.in));        int i,j,m,n;        i = in.nextInt();        j = in.nextInt();        BigInteger[][] A = new BigInteger[i][j];        for ( int a = 0 ; a < i ; a++){            for ( int b = 0 ; b < j ; b++){                A[a][b] = new BigInteger(in.next());            }        }        m = in.nextInt();        n = in.nextInt();        BigInteger[][] B = new BigInteger[m][n];        for ( int a = 0 ; a < m ; a++){            for ( int b = 0 ; b < n ; b++){                B[a][b] = new BigInteger(in.next());            }        }        BigInteger[][] C = Mul(A, j, B, n);        for ( int p = 0 ; p < C.length ; p++){            System.out.print(C[p][0]);            for ( int q = 1 ; q < n ; q++){                System.out.print(" "+C[p][q]);            }            System.out.println();        }        in.close();    }}

这里写图片描述

1 0
原创粉丝点击