您的位置:首页 > 其它

实现两个N*N矩阵的乘法,矩阵由一维数组表示

2011-08-13 16:48 316 查看
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

代码中,为了减少乘法次数,首先计算矩阵a,b,c的下标,然后在循环通过加法,更新a和b的下标。空间换时间,减少了计算下标所用的时间。

3. 备注

矩阵相乘是可以优化的,斯特拉森算法就可以把时间复杂度降低到n^(log7),不过原始算法是需要假设n是2的幂,而且实现中,需要分块,如果对一维数值进行分块的话,实现起来反而会增加时间和空间的负担。对于斯特拉森算法以后有机会再看了,今天就到这里。

4. 参考

矩阵相乘算法_C语言空间 /article/1455493.html
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: