给定一个序列 dims,其中 dims[i] 表示第 i 个矩阵的行数,dims[i+1]则是第i个矩阵的列数又同时是第i+1个矩阵的行数。矩阵连乘是指将这一系列矩阵相乘在一起,不同的乘法顺序会有不同的计算成本。你需要编写一个程序来找出一种最优的乘法顺序,使得总的乘法次数最少。
输入:
dims:一个整数列表,表示矩阵链的维度。例如,如果 dims = [10, 100, 5, 50],则表示有三个矩阵,第一个矩阵的维度是 10x100,第二个矩阵的维度是 100x5,第三个矩阵的维度是 5x50。
输出:
一个整数,表示最优乘法顺序下的最少乘法次数。
并用如下测试用例来验证:
测试用例1: dims = {10, 100, 5, 50}
测试用例2: dims = {30, 35, 15, 5, 10, 20, 25}
#include <stdio.h>
#include <stdlib.h>
#include <limits.h>
int matrixChainOrder(int *dims, int n) {
int **m = (int **)malloc((n - 1) * sizeof(int *));
int **s = (int **)malloc((n - 1) * sizeof(int *));
for (int i = 0; i < n - 1; i++) {
m[i] = (int *)malloc(n * sizeof(int));
s[i] = (int *)malloc(n * sizeof(int));
for (int j = 0; j < n; j++) {
m[i][j] = 0;
s[i][j] = 0;
}
}
for (int len = 2; len < n; len++) {
for (int i = 0; i < n - len; i++) {
int j = i + len;
m[i][j] = INT_MAX;
for (int k = i + 1; k < j; k++) {
int q = m[i][k] + m[k][j] + dims[i] * dims[k + 1] * dims[j];
if (q < m[i][j]) {
m[i][j] = q;
s[i][j] = k;
}
}
}
}
int result = m[0][n - 1];
for (int i = 0; i < n - 1; i++) {
free(m[i]);
free(s[i]);
}
free(m);
free(s);
return result;
}
int main() {
int dims1[] = { 10, 100, 5, 50 };
int n1 = sizeof(dims1) / sizeof(dims1[0]) - 1;
printf("测试用例1的最少乘法次数: %d\n", matrixChainOrder(dims1, n1));
int dims2[] = { 30, 35, 15, 5, 10, 20, 25 };
int n2 = sizeof(dims2) / sizeof(dims2[0]) - 1;
printf("测试用例2的最少乘法次数: %d\n", matrixChainOrder(dims2, n2));
return 0;
}