两个矩阵相乘的计算量
程序实现
//矩阵a和b相乘,p、q分别为a的行和列,q、r分别为b的行和列 void MatrixMultiply(int a[][MAXN], int b[][MAXN], int p, int q, int r) { int sum[MAXN][MAXN]; memset(sum, 0, sizeof(sum)); int i, j, k; //遍历矩阵a的行 for (k = 0; k < p; k++) { //遍历矩阵b的列 for (j = 0; j < r; j++) { //对应位置相乘 for (i = 0; i < q; i++) { sum[k][j] += a[k][i] * b[i][j]; } } } }
枚举所有完全加括号方式
ABCD四个矩阵连乘
程序实现
//m数组内存放矩阵链的行列信息 //m[i-1]和m[i]分别为第i个矩阵的行和列(i = 1、2、3...) int Best_Enum(int m[], int left, int right) { //只有一个矩阵时,返回计算次数0 if (left == right) { return 0; } int min = INF; //无穷大 int i; //括号依次加在第1、2、3...n-1个矩阵后面 for (i = left; i < right; i++) { //计算出这种完全加括号方式的计算次数 int count = Best_Enum(m, left, i) + Best_Enum(m, i+1, right); count += m[left-1] * m[i] * m[right]; //选出最小的 if (count < min) { min = count; } } return min; }
备忘录法优化
程序实现
#include <iostream> #include <cstdio> #include <cstring> using namespace std; #define SIZE 100 #define INF 999999999 int memo[SIZE][SIZE]; //m数组内存放矩阵链的行列信息 //m[i-1]和m[i]分别为第i个矩阵的行和列(i = 1、2、3...) int Best_Memo(int m[], int left, int right) { //只有一个矩阵时,返回计算次数0 if (left == right) { return 0; } int min = INF; int i; //括号依次加在第1、2、3...n-1个矩阵后面 for (i = left; i < right; i++) { //计算出这种完全加括号方式的计算次数 int count; if (memo[left][i] == 0) { memo[left][i] = Best_Memo(m, left, i); } count = memo[left][i]; if (memo[i+1][right] == 0) { memo[i+1][right] = Best_Memo(m, i+1, right); } count += memo[i+1][right]; count += m[left-1] * m[i] * m[right]; //选出最小的 if (count < min) { min = count; } } return min; } int main(void) { int m[SIZE]; int n; while (scanf("%d", &n) != EOF) { int i; for (i = 0; i < n; i++) { scanf("%d", &m[i]); } memset(memo, 0, sizeof(memo)); printf("%d\n", Best_Memo(m, 1, n-1)); } return 0; }
#include <iostream> #include <cstdio> #include <cstring> using namespace std; #define SIZE 100 #define INF 999999999 int m[SIZE]; //存放矩阵链的行列信息,m[i-1]和m[i]分别为第i个矩阵的行和列(i = 1、2、3...) int d[SIZE][SIZE]; //存放矩阵链计算的最优值,d[i][j]为第i个矩阵到第j个矩阵的矩阵链的最优值,i > 0 int Best_DP(int n) { //把d[i][i]置为0,1 <= i < n memset(d, 0, sizeof(d)); int len; //递归计算矩阵链的连乘最优值 //len = 1,代表矩阵链由两个矩阵构成 for (len = 1; len < n; len++) { int i, j, k; for (i = 1, j = i+len; j < n; i++, j++) { int min = INF; //无穷大 for (k = i; k < j; k++) { int count = d[i][k] + d[k+1][j] + m[i-1] * m[k] * m[j]; if (count < min) { min = count; } } d[i][j] = min; } } return d[1][n-1]; } int main(void) { int n; while (scanf("%d", &n) != EOF) { int i; for (i = 0; i < n; i++) { scanf("%d", &m[i]); } printf("%d\n", Best_DP(n)); } return 0; }