矩阵相乘---动态规划第一课

问题:给定n个矩阵{A1,A2,…,An},其中Ai与Ai+1是可乘的,i=1 ,…,n-1。如何确定计算矩阵连乘积的次序,使得依此次序计算矩阵连乘积需要的数乘次数最少。
在这里插入图片描述
在这里插入图片描述

#include <cstdio>
using namespace std;
const int MAXN = 100;
const int INF = 0x3f3f3f3f;
int dp[MAXN][MAXN]; // dp[i][j]表示矩阵Ai到Aj乘积的最少运算数
int idx[MAXN][MAXN]; // idx[i][j]表示矩阵Ai到Aj的划分位置
int a[MAXN]; // a[i]表示第i个矩阵的列数,a[i-1]表示第i个矩阵的行数
// 打印划分矩阵
int cnt = 1; // 划分次数
int pre[MAXN]; // 划分后,第i个矩阵的左括号数
int suf[MAXN]; // 划分后,第i个矩阵的右括号数
void divideMatrix(int i, int j)
{
    
    
    if (i == j) return;
    //i - j划分为 i - k, k + 1 - j;
    divideMatrix(i, idx[i][j]);
    divideMatrix(idx[i][j]+1, j);
    printf("第%d次计算为: (%d %d) - (%d %d)\n", cnt++, i, idx[i][j], idx[i][j]+1, j);
    pre[i]++;
    suf[j]++;
}

int main()
{
    
    
    int n;
    printf("请输入运算有多少个矩阵的乘积:\n");
    scanf("%d", &n);

    printf("请输入n+1个数代表矩阵的行列数:\n");
    for (int i = 0; i <= n; ++i)
    {
    
    
        scanf("%d", &a[i]);
    }

    // 初始化
    for (int i = 1; i <= n; ++i)
    {
    
    
        dp[i][i] = 0; 
    }
    // 从2个矩阵运算到n个矩阵
    for (int r = 2; r <= n; ++r)
    {
    
    
        // 从第i个矩阵开始计算r个矩阵 到 第n-r+1个矩阵开始计算r个矩阵
        for (int i = 1; i <= n - r + 1; ++i)
        {
    
    
            int j = i + r - 1;// 此时计算矩阵的最后一个矩阵,构成了Ai - Aj的矩阵序列
            dp[i][j] = INF;// 初始化为最大计算数
            idx[i][j] = i;

            for (int k = i; k < j; ++k)// 在i-j中寻找划分
            {
    
    
                int tmp = dp[i][k] + dp[k + 1][j] + a[i-1] * a[k] * a[j];
                if (tmp < dp[i][j]) // 遇到较小值更新
                {
    
    
                    dp[i][j] = tmp;
                    idx[i][j] = k;
                }
            }
        }
    }
    printf("矩阵乘积最小运算数是:%d\n\n", dp[1][n]);


    printf("矩阵进行以下运算:\n");
    divideMatrix(1, 6);
    for (int i = 1; i <= n; ++i)
    {
    
    
        for (int j = 1; j <= pre[i]; ++j)
        {
    
    
            printf("(");
        }
        printf("%d", i);
        for (int j = 1; j <= suf[i]; ++j)
        {
    
    
            printf(")");
        }
    }
}

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/WxqHUT/article/details/105730421