首页 > 代码库 > 矩阵连乘最优结合 动态规划求解

矩阵连乘最优结合 动态规划求解

1.引言  多矩阵连乘

对于一般的矩阵乘法来说,如矩阵A(m,n)与矩阵B(n,p)相乘需要进行的加法次数为m*n*p次乘法。

由于矩阵乘法满足结合律,因此矩阵相乘的结合性,会影响整个计算表达式的乘法执行次数。

如下面的例子,其中A(10,5)、B(5,20)、C(20,3):

    (1) ((AB)C) 执行乘法次数为1300次

    (2) (A(BC)) 执行乘法次数为450次

2.求最优的矩阵结合表达式

(1)设矩阵连乘积AiAi+1…Aj简记为A[i:j],设最优计算次序在Ak和Ak+1之间断开,则加括号方式为:

    ((AiAi+1…Ak) (Ak+1…Aj) )

则依照这个次序,先计算A[i:k]和A[k+1:j]然后再将计算结果相乘,计算量是:

    A[i:k]的计算量+A[K+1:j]的计算量+它们两者相乘的计算量

这里的关键是:计算A[i:j]的最优次序所包含的两个子过程(计算A[i:k]和A[K+1:j])也是最优次序

(2)具体计算

  设计算A[i,j]需要的乘法次数记为m[i,j]。

    M[i,j] = 0;    (i == j,表示一个矩阵,当然不需要乘法运算)

    M[i,j] = min(M[i,k]+M[k+1,j]+pi*pk*pj);   (k在[i,j)之间取值,表示分割点的位置,求最适合的分割点使得乘法次数最少)

  下面是使用动态规划计算6个矩阵连乘的示意图。可以使用自底向上计算,这样矩阵的分割点好计算。如先计算01两个矩阵乘积,在计算02三个矩阵乘积,在计算03四个矩阵乘积:

       01 12 23 34 45

       02 13 24 35

       03 14 25

       04 15

       05

 3.程序实例

程序可以根据给出的多个矩阵的行、列,生成最优结合的相乘表达式。

 1 #include <iostream> 2 #include <vector> 3 #include <algorithm> 4 #include <limits.h> 5 #include <string> 6 using namespace std; 7 ///计算M矩阵 8 int calculate_M(vector<vector<int> >&num,vector<pair<int,int> > &data,vector<vector<int> > &points){ 9     int len = data.size();10     for(int span = 1;span<len;span++){  ///间隔距离11         for(int col=0;col<len-span;col++){  ///操作起始列12 13             for(int i=col;i<col+span;i++){14                 int tmp = num[col][i] + num[i+1][col+span] + data[col].first*data[i].second*data[col+span].second;15                 if(tmp < num[col][col+span]){16                     points[col][col+span] = i;  ///记录分割点17                     num[col][col+span] = tmp;   ///记录最少乘法次数18                 }19             }20         }21     }22     return 0;23 }24 25 ///根据记录的分割点,生成最后的矩阵相乘表达式26 string make_result(vector<vector<int> > &points,int t1,int t2){27     if(t1 == t2)28         return string(1,A+t1);29     int split = points[t1][t2];30     return "("+make_result(points,t1,split)+"*"+make_result(points,split+1,t2)+")";31 }32 33 int main()34 {35     vector<pair<int,int>> data; ///保存矩阵的行、列36     data.push_back(make_pair(10,100));  //A37     data.push_back(make_pair(100,5));   //B38     data.push_back(make_pair(5,25));    //C39     data.push_back(make_pair(25,15));   //D40     data.push_back(make_pair(15,20));   //E41 42 43     int len = data.size();44     vector<vector<int> > num(len,vector<int>(len)); ///定义二维向量,并预先分配空间,记录乘法次数45     vector<vector<int> > points(len,vector<int>(len)); ///定义二维向量,并预先分配空间,记录分割点46     for(int i=0;i<len;i++){47         for(int j=0;j<len;j++){48             points[i][j] = -1;49             if(i == j)50                 num[i][j] = 0;  ///自己和自己相乘,所以为051             else52                 num[i][j] = INT_MAX;    ///否则,记为最大整数值53         }54     }55 56     calculate_M(num,data,points);57     cout<<make_result(points,0,len-1)<<"\t最少乘法次数为:"<<num[0][len-1]<<endl;58     return 0;59 }
View Code

输入矩阵,表示每个矩阵的行、列:

输出最优的结合表达式:

  

矩阵连乘最优结合 动态规划求解