实现两个N*N矩阵的乘法,矩阵由一维数组表示
1. 简述
注意下标、参数传递。只要把二维中下标,转化为一维数组的下标即可。
2. 核心代码
二维的公式是:c[i][j] += a[i][k] * b[k][j], k=0,1,2,...,N-1
一维的公式: c[i*N+j] += a[i*N+k] * b[k*N+j], k=0,1,2,...,N-1
template<class T>
void MaxtrixMultiply(const T* a,const T* b, T* c, unsigned int N) {
for(int i=0; i<N; i++) {
for(int j=0; j<N; j++) {
int c_index = i*N+j;
int a_index = i*N+0;
int b_index = 0*N+j;
c[c_index] = 0;
for(int k=0; k<N; k++) {
c[c_index] += a[a_index]*b[b_index];
a_index += 1;
b_index += N;
} // for
} // for
} // for
} // MaxtrixMultiply
void MaxtrixMultiply(const T* a,const T* b, T* c, unsigned int N) {
for(int i=0; i<N; i++) {
for(int j=0; j<N; j++) {
int c_index = i*N+j;
int a_index = i*N+0;
int b_index = 0*N+j;
c[c_index] = 0;
for(int k=0; k<N; k++) {
c[c_index] += a[a_index]*b[b_index];
a_index += 1;
b_index += N;
} // for
} // for
} // for
} // MaxtrixMultiply
代码中,为了减少乘法次数,首先计算矩阵a,b,c的下标,然后在循环通过加法,更新a和b的下标。空间换时间,减少了计算下标所用的时间。
3. 备注
矩阵相乘是可以优化的,斯特拉森算法就可以把时间复杂度降低到n^(log7),不过原始算法是需要假设n是2的幂,而且实现中,需要分块,如果对一维数值进行分块的话,实现起来反而会增加时间和空间的负担。对于斯特拉森算法以后有机会再看了,今天就到这里。
4. 参考
矩阵相乘算法_C语言空间 http://blog.csdn.net/chief1985/article/details/2402553