很久没写blog了,感觉人都快变的抑郁了,换工作之后各种揪心,说好了是做Android的,结果让我搞各种算法,也罢,权当学习了一点知识吧。
今天说说矩阵相乘的算法,计算算法很简单,就是3个for循环。
首先还是说下矩阵相乘的概念,其实大学的时候线性代数中应该有讲到,不过到现在估计都还给老师了。
废话不多说,矩阵,其实就是一个二维数组,横竖排列的,比如int[5][6],就是一个矩阵,表示有5行6列。
只有当矩阵A的列数与矩阵B的行数相等时A×B才有意义。一个m×n的矩阵a(m,n)左乘一个n×p的矩阵b(n,p),会得到一个m×p的矩阵c(m,p)。左乘:又称前乘,就是乘在左边(即乘号前),比如说,A左乘E即AE。
在计算机中,一个矩阵实际上就是一个二维数组。一个m行n列的矩阵与一个n行p列的矩阵可以相乘,得到的结果是一个m行p列的矩阵,其中的第i行第j列位置上的数为第一个矩阵第i行上的n个数与第二个矩阵第j列上的n个数对应相乘后所得的n个乘积之和。比如,下面的算式表示一个2行2列的矩阵乘以2行3列的矩阵,其结果是一个2行3列的矩阵。
算法:
1 //矩阵相乘
2 public static float[][] Mul(float[][] a, float[][] b) {
3 //确保矩阵a的列数和b的行数相等
4 if(a[0].length != b.length) {
5 return null;
6 }
7 //用来存放结果的矩阵,axb的结果为a的行数和b的列数
8 float[][] result = new float[a.length][b[0].length];
9 //对a的每行进行遍历
10 for(int i=0; i<a.length; i++) {
11 //对b的每列进行遍历
12 for(int j=0;j<b[0].length; j++) {
13 //c为每一个点的值
14 float c = 0;
15 //第i行j列的值为a的第i行上的n个数和b的第j列上的n个数对应相乘之和,其中n为a的列数,也是b的行数,a的列数和b的行数相等
16 for(int k=0; k<a[0].length; k++) {
17 c += (a[i][k]*b[k][j]);
18 }
19 result[i][j] = c;
20 }
21 }
22 return result;
23 }
代码注释的很清楚了,主要是抓住定义,3个for循环。如果你的二维数组不是float类型,可以相应的更改,记得将c和返回值一并更改。
就到这里吧。