【算法笔记】动态规划:矩阵连乘问题

连乘次数

A是一个p\times q矩阵,B是一个q\times r矩阵,AB相乘,得到的矩阵元素个数为p\times r,每个元素由q次乘法得到,因此所需乘法次数为p\times q\times r


问题描述

在计算矩阵连乘积时,加括号的方式对计算量有影响。

例如有三个矩阵A_1,A_2,A_3连乘,它们的维数分别为
10\times100,100\times5,5\times50。用第一种加括号方式(A_1A_2)A_3计算,则所需数乘次数为10\times100\times5+10\times5\times50=7,500。用第二种加括号方式A_1(A_2A_3)计算,需要100\times5\times50+10\times100\times50=75,000次数乘。

输入连乘矩阵的个数,每个矩阵的维数。要求输出数乘次数最少时的加括号方式,及数乘次数。

Sample Input

6
30 35
35 15
15 5
5 10
10 20
20 25

Sample Output

15125
((A1(A2A3))((A4A5)A6))


解题思路:

先引入以下符号:

  • n表示矩阵的个数
  • A_i表示第i个矩阵
  • A[i:j]表示矩阵连乘A_iA_{i+1}..A_j
  • p_i表示A_i的列数
  • p_{i-1}表示A_i的行数
  • k表示矩阵连乘断开的位置为k,表示在A_kA_{k+1}之间断开
  • m[i,j]表示A[i:j]的最少乘次,m[1,n]即问题的最优解

符号解释:由矩阵相乘的条件可知:前一个矩阵的列数 = 后一个矩阵的行数。因此,p_i既是A_i的列数,也是A_{i+1}的行数。结合下面的示意图有助于理解:

该问题有一个关键特征:计算矩阵链A[1:n]的最优计算方式,包含了子矩阵链A[1:k]A[k+1:n]的最优计算方式。

一般地,如果原问题的最优解,包含了其子问题的最优解,则我们称这种性质为最优子结构性质。若问题具有最优子结构性质,则可用动态规划算法求解。


首先,建立递归关系,写出递归公式

m[i,j]=\begin{cases} 0, & i=j \\ \min\limits_{i\leqslant k < j}(m[i,k]+m[k+1,j]+p_{i-1}p_kp_j ), & i<j \end{cases}

公式解释:假设k是对矩阵链A[i:j]一分为二得到最优解时的断开位置,则m[i,k]m[k+1,j]分别是两个子矩阵链A[i,k]A[k+1,j]的最优解。两个矩阵最后要相乘才能得到A[i,j],因此,最后要加上p_{i-1}p_kp_j,也就是两子矩阵相乘的数乘次数,才能得到总的数乘次数。


然后,根据递归公式计算最优解

在程序中,m的实现是一个二维数组,也就是一张二维表,为了方便计算,m的下标从1开始。要计算A[1:n]的最少乘次,本质上是求m[1][n]的值,也就是二维表表右上角的值.

例如,连乘矩阵个数为6,维数分别为:
A1(30\times35),
A2(35\times15),
A3(15\times5),
A4(5\times10),
A5(10\times20),
A6(20\times25)

填表方式如下图所示:

原问题现已转化为填表问题,要填充的是矩阵右上三角部分的值。
根据递归公式,对角线的值为0。其他值需要根据于断开位置k的值来得到,k \in [i,j),我们要遍历所有k,就要访问所求值的所有同一行左边的值和同一列下方的值。因此,在代码中我们可以使用自底向上、从左到右的计算顺序来依次填充,最终得到右上角的值。

例如:
m[2][5]{=\min\begin{cases} m[2][2]+m[3][5]+p_1p_2p_5=13000 \\ m[2][3]+m[4][5]+p_1p_3p_5=7125 \\ m[2][4]+m[5][5]+p_1p_4p_5=11375 \end{cases}}=7125\\

伪代码如下:

for (int i = n; i >= 1; i--) //i表示行
{
    for (int j = i; j <= n; j++) //j表示列
    {
        if (i == j) 
            m[i][j] = 0;
        else
            m[i][j] = CalculateMin(i,j);
    }
}

完整代码

//矩阵连乘问题,递归方程采用课本P47的递归方程。
//与课本P47的程序相比,两者都是填m[][]的上三角,
//但本程序是横向填表,而课本程序是斜向填表。
//本程序按照 自底向上,自左向右 的顺序来计算m[][]的上三角。
//程序中有记录相应断开位置到s[][]。

#include <iostream>
using namespace std;

void MatrixChain(int, int[], int **, int **);
void print(int, int, int **);
void CalculateMin(int i, int j, int p[], int **m, int **s);

int main()
{
    int n;
    cin >> n;
    int *tempP = new int[2 * n]; //p[0]为A1的行数;p[i]为Ai的列数

    int *p = new int[n + 1];    //p[0]为A1的行数;p[i]为Ai的列数
    int **m = new int *[n + 1]; //m[i][j]为Ai连乘到Aj的最少乘次数
    int **s = new int *[n + 1]; //s[i][j]为 Ai连乘到Aj的最优解的第一层断开位置
    for (int i = 1; i <= n; i++)
    {
        m[i] = new int[n + 1];
        s[i] = new int[n + 1];
    }

    for (int i = 0; i < 2 * n; i++)
    {
        cin >> tempP[i];
    }

    for (int i = 0; i <= n; i++)
    {
        p[i] = i == 0 ? tempP[0] : tempP[i * 2 - 1];
    }

    MatrixChain(n, p, m, s);
    cout << m[1][n] << endl; //最优值存储在m[1][n]
    //print(1, n, s); //打印最优计算方式

    return 0;
}

void MatrixChain(int n, int p[], int **m, int **s)
{
    //以下双重循环对m[][]的上三角进行填表
    //填表顺序为自底向上,自左向右
    for (int i = n; i >= 1; i--) //i表示行
    {
        for (int j = i; j <= n; j++) //j表示列,因为只填上三角,所以j的初值为i
        {
            //以下按照矩阵连乘问题的递归公式来求每一个m[i][j]
            if (i == j)
            {
                m[i][j] = 0;
            }
            else
            {
                CalculateMin(i, j, p, m, s);
            }
        }
    }
}

void CalculateMin(int i, int j, int p[], int **m, int **s)
{
    m[i][j] = m[i][i] + m[i + 1][j] + p[i - 1] * p[i] * p[j]; //若i与j不相等,m[i][j]的初值为断开位置为i时的最优值

    s[i][j] = i;                    //记录当前断开位置位置为i
    for (int k = i + 1; k < j; k++) //k用于尝试不同的断开位置
    {
        int curr = m[i][k] + m[k + 1][j] + p[i - 1] * p[k] * p[j];
        if (curr < m[i][j]) //当前k值作为Ai连乘到Aj的断开位置能获得更少计算量
        {
            m[i][j] = curr; //刷新最优值
            s[i][j] = k;   //刷新相应断开位置
        }
    }
}

void print(int i, int j, int **s)
{
    if (i == j)
        cout << "A" << i;
    else
    {
        cout << "(";
        print(i, s[i][j], s);
        print(s[i][j] + 1, j, s);
        cout << ")";
    }
}

可以在这个网站上检验一下程序的正确性:https://onlinejudge.u-aizu.ac.jp/problems/ALDS1_10_B


参考资料:《计算机算法设计与分析》(第四版)p44-p49

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 198,932评论 5 466
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,554评论 2 375
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 145,894评论 0 328
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,442评论 1 268
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,347评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 47,899评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,325评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,980评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,196评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,163评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,085评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,826评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,389评论 3 302
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,501评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,753评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,171评论 2 344
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,616评论 2 339

推荐阅读更多精彩内容