算法提高 矩阵相乘

原创 2017年02月12日 15:22:33

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

  现在给你一个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();
    }

}

这里写图片描述

版权声明:本文为博主原创文章,未经博主允许不得转载。若需转载,请注明http://blog.csdn.net/qq_30091945 举报

相关文章推荐

算法提高 矩阵相乘

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

算法提高 矩阵相乘

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

我是如何成为一名python大咖的?

人生苦短,都说必须python,那么我分享下我是如何从小白成为Python资深开发者的吧。2014年我大学刚毕业..

算法提高 矩阵乘法

问题描述   有n个矩阵,大小分别为a0*a1, a1*a2, a2*a3, ..., a[n-1]*a[n],现要将它们依次相乘,只能使用结合率,求最少需要多少次运算。   两个大小分别为p...

算法提高 矩阵乘法

问题描述   有n个矩阵,大小分别为a0*a1, a1*a2, a2*a3, ..., a[n-1]*a[n],现要将它们依次相乘,只能使用结合率,求最少需要多少次运算。   两个大小分别为p*q和q...

Strassen矩阵相乘算法

引言 Strassen的矩阵相乘方法是一种典型的分治算法。目前为止,我们已经见过一些分治策略的算法了,例如归并排序和Karatsuba大数快速乘法。现在,让我再来看看分治策略的背后是什么。 同动态...

Strassen矩阵相乘算法

引言Strassen的矩阵相乘方法是一种典型的分治算法。目前为止,我们已经见过一些分治策略的算法了,例如归并排序和Karatsuba大数快速乘法。现在,让我再来看看分治策略的背后是什么。同动态规划不同...

提高项目34-矩阵相乘

#include int main() { int a[2][3] = {1,1,0,2,0,3}; int b[3][4] = {0,2,3,1,1,0,2,2,2,1,1,1}...

矩阵相乘算法——Strassen算法

引言 Strassen的矩阵相乘方法是一种典型的分治算法。目前为止,我们已经见过一些分治策略的算法了,例如归并排序和Karatsuba大数快速乘法。现在,让我再来看看分治策略的背后是什么。 ...

NYOJ 623 A*B Problem II

A*B Problem II 时间限制:1000 ms  |  内存限制:65535 KB 描述 ACM的C++同学有好多作业要做,最头痛莫过于线性代数了,因为每次做到矩阵相乘的...

PM3(矩阵相乘 行和的简便运算)

http://poj.org/problem?id=3213 PM3 Time Limit: 5000MS   Memory Limit: 131072K T...
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)