STRASSEN矩阵乘法算法.docVIP

STRASSEN矩阵乘法算法 //STRASSEN矩阵乘法算法 #include iostream.h const int N=8; //常量N用来定义矩阵的大小 void main() { void STRASSEN(int n,float A[][N],float B[][N],float C[][N]); void input(int n,float p[][N]); void output(int n,float C[][N]); //函数声明部分 float A[N][N],B[N][N],C[N][N]; //定义三个矩阵A,B,C cout现在录入矩阵A[N][N]:endlendl; input(N,A); coutendl现在录入矩阵B[N][N]:endlendl; input(N,B); //录入数组 STRASSEN(N,A,B,C); //调用STRASSEN函数计算 output(N,C); //输出计算结果 } void input(int n,float p[][N]) //矩阵输入函数 { int i,j; for(i=0;in;i++) { cout请输入第i+1行endl; for(j=0;jn;j++) cinp[i][j]; } } void output(int n,float C[][N]) //据矩阵输出函数 { int i,j; cout输出矩阵:endl; for(i=0;in;i++) { coutendl; for(j=0;jn;j++) coutC[i][j] ; } coutendlendl; } void MATRIX_MULTIPLY(float A[][N],float B[][N],float C[][N]) //按通常的矩阵乘法计算C=AB的子算法(仅做2阶) { int i,j,t; for(i=0;i2;i++) //计算A*B--C for(j=0;j2;j++) { C[i][j]=0; //计算完一个C[i][j]，C[i][j]应重新赋值为零 for(t=0;t2;t++) C[i][j]=C[i][j]+A[i][t]*B[t][j]; } } void MATRIX_ADD(int n,float X[][N],float Y[][N],float Z[][N]) //矩阵加法函数X+Y—Z { int i,j; for(i=0;in;i++) for(j=0;jn;j++) Z[i][j]=X[i][j]+Y[i][j]; } void MATRIX_SUB(int n,float X[][N],float Y[][N],float Z[][N]) //矩阵减法函数X-Y—Z { int i,j; for(i=0;in;i++) for(j=0;jn;j++) Z[i][j]=X[i][j]-Y[i][j]; } void STRASSEN(int n,float A[][N],float B[][N],float C[][N]) //STRASSEN函数（递归） { float A11[N][N],A12[N][N],A21[N][N],A22[N][N]; float B11[N][N],B12[N][N],B21[N][N],B22[N][N]; float C11[N][N],C12[N][N],C21[N][N],C22[N][N]; float M1[N][N],M2[N][N],M3[N][N],M4[N][N],M5[N][N],M6[N][N],M7[N][N]; float AA[N][N],BB[N][N],MM1[N][N],MM2[N][N]; int i,j;//,x; if (n==2) MATRIX_MULTIPLY(A,B,C);//按通常的矩阵乘法计算C=AB的子算法(仅做2阶) else { for(i=0;in/2;i++) ////////// for(j=0;jn/2;j++) { A11[i][j]=A[i][j]; A12[i][j]=A[i][j+n/2]; A21[i][j]=A[i+n/