纠结犹豫了几天,终于还是下定决心写一下矩阵乘法。
矩阵乘法有三种实现方法:暴力求解法、分治法、Strassen方法。
暴力求解法就是用for循环,一个一个元素的算;
分治法是把矩阵分为四块儿,再递归算;
Strassen方法则是根据公式来计算。
其实分治法和Strassen方法调用的函数是差不多的,但是Strassen方法的时间复杂度要小很多。
下面就是源码了:
multiply.h
#include<stdio.h>
#include<stdlib.h>
void mult(int **a,int **b,int **c,int n);//暴力求解法
void multr(int **a,int **b,int **c,int n);//简单分治法
void mults(int **a,int **b,int **c,int n);//Strassen方法
multiply.cpp
#include"multiply.h"
void divide(int **a,int **a11,int **a12,int **a21,int **a22,int n);
void add(int **a,int **b,int **c,int n);
void unit(int **a,int **a11,int **a12,int **a21,int **a22,int n);
void sub(int **a,int **b,int **c,int n);
void divide(int **a,int **a11,int **a12,int **a21,int **a22,int n){//把矩阵分为四块儿
int i,j;
for(i=0;i<n;i++)
for(j=0;j<n;j++){
a11[i][j]=a[i][j];
a12[i][j]=a[i][j+n];
a21[i][j]=a[i+n][j];
a22[i][j]=a[i+n][j+n];
}
}
void add(int **a,int **b,int **c,int n){//矩阵加法
int i,j;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
c[i][j]=a[i][j]+b[i][j];
}
void sub(int **a,int **b,int **c,int n){//矩阵减法
int i,j;
for(i=0;i<n;i++)
for(j=0;j<n;j++)
c[i][j]=a[i][j]-b[i][j];
}
void unit(int **a,int **a11,int **a12,int **a21,int **a22,int n){//把四小块儿矩阵合并为一个矩阵
int i,j;
for(i=0;i<n;i++)
for(j=0;j<n;j++){
a[i][j]=a11[i][j];
a[i][j+n]=a12[i][j];
a[i+n][j]=a21[i][j];
a[i+n][j+n]=a22[i][j];
}
}
void mult(int **a,int **b,int **c,int n){//暴力求解