矩阵相乘（分治法）

最新推荐文章于 2024-03-20 10:58:55 发布

原创

最新推荐文章于 2024-03-20 10:58:55 发布 · 1.2w 阅读

66 ·

CC 4.0 BY-SA版权

文章标签：

#c语言 #分治算法 #递归 #矩阵相乘 #下标计算

一个简单的分治算法求矩阵相乘
C=A * B ，假设三个矩阵均为n×n，n为2的幂。可以对其分解为4个n/2×n/2的子矩阵分别递归求解：

递归分治算法：

算法中一个重要的细节就是在分块的时候，采用的是下标的方式。

#include <stdio.h>
#include <stdlib.h>
#define ROW 16       //指定 行数
#define COL 16       //指定 列数 

int a[ROW][COL],b[ROW][COL];  //矩阵a 和 矩阵b
int **c;                      // c = a * b 

//保存一个矩阵的第一个元素的位置，即左上角元素的下标
//如果加上一个长度就可以知道整个矩阵了
typedef struct {   //这里没有指定一个矩阵的长度，在分块时应该加入长度，否则不知道子块矩阵的大小
    int str,stc;    //str行下标  ； strc列下标
}subarr;

// 两矩阵arr、brr相加减 保存在temp中
void operate(int **arr,int **brr,subarr te,char op,int **temp,int len);

//分治法 求矩阵相乘 ，sa,sb分别为矩阵a,b参加运算的首元素
int ** square_recursive(subarr sa,subarr sb,subarr sc,int len){
    int n=len;
    int **temp;
    int i;
    // 申请一个临时矩阵，用于保存a*b 
    temp=(int**)malloc(sizeof(int *)*n);
    for ( i=