矩阵连乘问题

给定一个序列 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;
}

  • 4
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值