Conquer and Divide经典例子之Strassen算法解决大型矩阵的相乘
在通过汉诺塔问题理解递归的精髓中我讲解了怎么把一个复杂的问题一步步recursively划分了成简单显而易见的小问题。其实这个解决问题的思路就是算法中常用的divide and conquer, 这篇日志通过解决矩阵的乘法,来了解另外一个基本divide and conque思想的strassen算法。
矩阵A乘以B等于X, 则Xij =
注意左乘右乘的区别,AB 与BA是不同的。
如果r = 1, 直接就是两个数的相乘。
如果r = 2, 例如
X =
[ 1, 2;
3, 4];
Y =
[ 2, 3;
4, 5];
R = XY的计算十分简单,但是如果r很大,耗时是O(r^3)。为了简化,可以把X, Y各自划分成2X2的矩阵,每一个元素其实是有n/2行的矩阵
(注:这里仅讲解行数等于列数的情况。)
X =
[A, B;
C, D];
Y =
[E, F;
G, H]
所以XY =[
AE+BG, AF+BH;
CE+DG, CF+DH]
Strassen引入seven magic product 分别是P1, P2, P3 ,P4, P5, P6, P7
P1 = A(F-H)
P2 = (A+B)H
P3 = (C+D)E
P4 = D(G-E)
P5 = (A+D)(E+H)
P6 = (B-D)(G+H)
P7 = (A-C)(E+F)
这样XY =
[P5+P4-P2+P6, P1+P2;
P3+P4, P1+P5-P3-P7]
然后通过递归的策略计算矩阵的相乘,递归的出口是n = 1.
关键点就是这些,附上代码吧。
- //multiply matrix multiplication
- import java.util.Scanner;
- public class Strassen{
- public Strassen(){}
- /** split a parent matrix into child matrics8*/
- public static void split(int[][] P, int[][] C, int iB, int jB){
- for(int i1=0, i2 = iB; i1<C.length; i1++, i2++)
- for(int j1=0, j2=jB; j1<C.length; j1++, j2++)
- C[i1][j1] = P[i2][j2];
- }
- /**join child matric into parent matrix*/
- public static void join(int[][] C, int[][] P, int iB, int jB){
- for(int i1=0, i2 = iB; i1<C.length; i1++, i2++)
- for(int j1=0, j2=jB; j1<C.length; j1++, j2++)
- P[i2][j2]=C[i1][j1];
- }
- /**add two matrics into one*/
- public static int[][] add(int[][] A, int[][] B){
- //A and B has the same dimension
- int n = A.length;
- int[][] C = new int[n][n];
- for (int i=0; i<n; i++)
- for(int j=0; j<n; j++)
- C[i][j] = A[i][j] + B[i][j];
- return C;
- }
- //subtract one matric by another
- public static int[][] sub(int[][] A, int[][] B){
- //A and B has the same dimension
- int n = A.length;
- int[][] C = new int[n][n];
- for (int i=0; i<n; i++)
- for(int j=0; j<n; j++)
- C[i][j] = A[i][j] - B[i][j];
- return C;
- }
- //Multiply matrix
- public static int[][] multiply(int[][] A, int[][] B){
- int n = A.length;
- int[][] R = new int[n][n];
- /**exit*/
- if(n==1)
- R[0][0] = A[0][0]+B[0][0];
- else{
- //divide A into 4 submatrix
- int[][] A11 = new int[n/2][n/2];
- int[][] A12 = new int[n/2][n/2];
- int[][] A21 = new int[n/2][n/2];
- int[][] A22 = new int[n/2][n/2];
- split(A, A11, 0, 0);
- split(A, A12, 0, n/2);
- split(A, A21, n/2, 0);
- split(A, A22, n/2, n/2);
- //divide B into 4 submatric
- int[][] B11 = new int[n/2][n/2];
- int[][] B12 = new int[n/2][n/2];
- int[][] B21 = new int[n/2][n/2];
- int[][] B22 = new int[n/2][n/2];
- split(B, B11, 0, 0);
- split(B, B12, 0, n/2);
- split(B, B21, n/2, 0);
- split(B, B22, n/2, n/2);
- //seven magic products
- int[][] P1 = multiply(A11, sub(B12, B22));
- int[][] P2 = multiply(add(A11,A12), B22);
- int[][] P3 = multiply(add(A21, A22), B11);
- int[][] P4 = multiply(A22, sub(B21, B11));
- int[][] P5 = multiply(add(A11, A22), add(B11, B22));
- int[][] P6 = multiply(sub(A12, A22), add(B21, B22));
- int[][] P7 = multiply(sub(A11, A21), add(B11, B12));
- //new 4 submatrix
- int[][] R11 = add(add(P5, sub(P4, P2)), P6);
- int[][] R12 = add(P1, P2);
- int[][] R21 = add(P3, P4);
- int[][] R22 = sub(sub(add(P1, P5), P3), P7);
- //joint together
- join(R11, R, 0, 0);
- join(R12, R, 0, n/2);
- join(R21, R, n/2, 0);
- join(R22, R, n/2, n/2);
- }
- return R;
- }
- //main
- public static void main(String[] args){
- Scanner scan = new Scanner(System.in);
- System.out.println("Strassen Multiplication Algorithm Test\n");
- Strassen s = new Strassen();
- System.out.println("Fetch the matric A and B...");
- int N = scan.nextInt();
- int[][] A = new int[N][N];
- int[][] B = new int[N][N];
- for (int i = 0; i < N; i++)
- for (int j = 0; j < N; j++)
- A[i][j] = scan.nextInt();
- for (int i = 0; i < N; i++)
- for (int j = 0; j < N; j++)
- B[i][j] = scan.nextInt();
- System.out.println("Fetch Completed!");
- int[][] C = s.multiply(A, B);
- System.out.println("\nmatrices A = ");
- for (int i = 0; i < N; i++){
- for (int j = 0; j < N; j++)
- System.out.print(A[i][j] +" ");
- System.out.println();
- }
- System.out.println("\nmatrices B =");
- for (int i = 0; i < N; i++) {
- for (int j = 0; j < N; j++)
- System.out.print(B[i][j] +" ");
- System.out.println();
- }
- System.out.println("\nProduct of matrices A and B = ");
- for (int i = 0; i < N; i++)
- {
- for (int j = 0; j < N; j++)
- System.out.print(C[i][j] +" ");
- System.out.println();
- }
- }
- }
Conquer and Divide经典例子之Strassen算法解决大型矩阵的相乘的更多相关文章
- Conquer and Divide经典例子之汉诺塔问题
递归是许多经典算法的backbone, 是一种常用的高效的编程策略.简单的几行代码就能把一团遭的问题迎刃而解.这篇博客主要通过解决汉诺塔问题来理解递归的精髓. 汉诺塔问题简介: 在印度,有这么一个古老 ...
- Strassen算法
如题,该算法是来自德国的牛逼的数学家strassen搞出来的,因为把n*n矩阵之间的乘法复杂度降低到n^(lg7)(lg的底是2),一开始想当然地认为朴素的做法是n^3,哪里还能有复杂度更低的做法,但 ...
- 4-2.矩阵乘法的Strassen算法详解
题目描述 请编程实现矩阵乘法,并考虑当矩阵规模较大时的优化方法. 思路分析 根据wikipedia上的介绍:两个矩阵的乘法仅当第一个矩阵B的列数和另一个矩阵A的行数相等时才能定义.如A是m×n矩阵和B ...
- 第四章 分治策略 4.2 矩阵乘法的Strassen算法
package chap04_Divide_And_Conquer; import static org.junit.Assert.*; import java.util.Arrays; import ...
- C/C++中几种经典的垃圾回收算法
1.引用计数算法 引用计数(Reference Counting)算法是每个对象计算指向它的指针的数量,当有一个指针指向自己时计数值加1:当删除一个指向自己的指针时,计数值减1,如果计数值减为0,说明 ...
- python之路第五篇之递归(进阶篇:续:经典例子剖析)
递归 在函数内部,可以调用其他函数; 如果一个函数在内部调用自身本身,这个函数就是递归函数. 例如,我们来计算阶乘: n! = 1 x 2 x 3 x ... x n, 用函数f1(n)表示,可以看出 ...
- jQuery监听事件经典例子
关键字:jQuery监听事件经典例子 js代码: ============================================================ $(function( ...
- 记录几个经典的字符串hash算法
记录几个经典的字符串hash算法,方便以后查看: 推荐一篇文章: http://www.partow.net/programming/hashfunctions/# (1)暴雪字符串hash #inc ...
- C语言经典算法 - 多维矩阵转一维矩阵的代码
下边内容内容是关于C语言经典算法 - 多维矩阵转一维矩阵的内容,应该能对码农也有好处. #include <stdio.h>#include <stdlib.h>int mai ...
随机推荐
- 探秘Tomcat——连接篇
前两篇我们分别粗线条和细粒度的讲解了tomcat的服务是如何启动以及连接器Connector和容器Container又分别是如何被启动的. 本篇我们主要侧重tomcat中server.service以 ...
- 去百度API的百度地图准确叠加和坐标转换的解决方案研究
文章版权由作者李晓晖和博客园共有,若转载请于明显处标明出处:http://www.cnblogs.com/naaoveGIS/. 1.背景 目前项目上如果要使用百度地图,得加载百度的开发包,然后通过百 ...
- 【JUC】JDK1.8源码分析之AbstractQueuedSynchronizer(二)
一.前言 在锁框架中,AbstractQueuedSynchronizer抽象类可以毫不夸张的说,占据着核心地位,它提供了一个基于FIFO队列,可以用于构建锁或者其他相关同步装置的基础框架.所以很有必 ...
- Android系统build.prop文件
# begin build properties (开始设置系统性能) # autogenerated by buildinfo.sh (通过设置形成系统信息) ro.build.id=GRI40 ( ...
- CQRS, Task Based UIs, Event Sourcing agh!
原文地址:CQRS, Task Based UIs, Event Sourcing agh! Many people have been getting confused over what CQRS ...
- 图片上传插件ImgUploadJS:用HTML5 File API 实现截图粘贴上传、拖拽上传
一 . 背景及效果 当前互联网上传文件最多的就是图片文件了,但是传统web图片的截图上传需要:截图保存->选择路径->保存后再点击上传->选择路径->上传->插入. 图片 ...
- 设计模式(Design Pattern)系列之.NET专题
最近,不是特别忙,重新翻了下设计模式,特地在此记录一下.会不定期更新本系列专题文章. 设计模式(Design pattern)是一套被反复使用.多数人知晓的.经过分类编目的.代码设计经验的总结. 使用 ...
- ASP.NET MVC使用Areas后怎样获取Area(区域)的名称
写此随笔,目的只为今后在ASP.NET MVC项目中再用到Area(区域)时作为备查. 获取当前Area(区域)名称的方法是: ViewContext.RouteData.DataTokens[&qu ...
- C#创建目录,文件名过滤特殊字符串,非法字符
string invalid = new string(Path.GetInvalidFileNameChars()) + new string(Path.GetInvalidPathChars()) ...
- 批量导出access某表内容到word文档
一.需求: 需要将表中每一条记录中的某些内容导出在一个word文档中,并将这些文档保存在指定文件夹目录下 二.界面,简单设计如下: 三.添加office相关引用 添加后可在解决方案资源管理器中看到: ...