首页 > 代码库 > 算法之矩阵连乘

算法之矩阵连乘

一.问题描叙

    给定n个矩阵{A1,A2,……,An},其中Ai与Ai+1是可乘的,i=1,2,……,n-1。

   例如:

     计算三个矩阵连乘{A1,A2,A3};维数分别为10*100 , 100*5 , 5*50

     按此顺序计算需要的次数((A1*A2)*A3):10X100X5+10X5X50=7500次

     按此顺序计算需要的次数(A1*(A2*A3)):10X5X50+10X100X50=75000次

     所以要解决的问题是:如何确定矩阵连乘积A1A2,……An的计算次序,使得按此计算次序计算矩阵连乘积需要的数乘次数达到最小化。

二.问题分析

   由于矩阵乘法满足结合律,所以计算矩阵连乘的连乘积可以与许多不同的计算计算次序,这种计算次序可以用加括号的方式来确定。若一个矩阵连乘积的计算次序完全确定,也就是说连乘积已完全加括号,那么可以依此次序反复调用2个矩阵相乘的标准算法计算出矩阵连乘积。

   完全加括号的矩阵连乘积可递归地定义为:

   (1).单个矩阵是完全加括号的;

  (2).矩阵连乘积A是完全加括号的,则A可以表示为2个完全加括号的矩阵连乘积B和C的乘积并加括号,及A=(BC);

    举个例子,矩阵连乘积A1A2A3A4A5,可以有5种不同的完全加括号方式:

       (A1(A2(A3A4))),(A1((A2A3)A4)),((A1A2)(A3A4)),((A1(A2A3))A4),(((A1A2)A3)A4)

    每一种完全加括号的方式对应一种矩阵连乘积的计算次序,而矩阵连乘积的计算次序与其计算量有密切的关系,即与矩阵的行和列有关。

    补充一下数学知识,矩阵A与矩阵B可乘的条件为矩阵A的列数等于矩阵B的行数,例如,若A是一个p*q的矩阵,B是一个q*r的矩阵,则其乘积C=AB是一个p*r的矩阵。

三.动态规划解决矩阵连乘积的最优计算次序问题

    或许你对动态规划有点陌生,那简单的讲讲什么叫动态规划吧。

    动态规划算法与分治法类似,其基本思想也就是将待求解的问题分解成若干个子问题,先求解子问题,然后从这些子问题的解得到原问题的解,简单概括为自顶向下分解,自底向上求解。与分治法不同的是,适合于用动态规划法求解的问题,经分解得到的子问题往往不是相互独立的,换句话说,就是前面解决过的子问题,在后面的子问题中又碰到了前面解决过的子问题,子问题之间是有联系的。如果用分治法,有些同样的子问题会被重复计算几次,这样就很浪费时间了。所以动态规划是为了解决分治法的弊端而提出的,动态规划的基本思想就是,用一个表来记录所有已经解决过的子问题的答案,不管该子问题在以后是否会被用到,只要它被计算过,就将其结果填入表中,以后碰到同样的子问题,就可以从表中直接调用该子问题的答案,而不需要再计算一次。具体的动态规划的算法多种多样,但他们都具有相同的填表式。

   顺便说一下动态规划的适用场合,一般适用于解最优化问题,例如矩阵连乘问题、最长公共子序列、背包问题等等,通常动态规划的设计有4个步骤,结合矩阵连乘分析:

   (1).找出最优解的性质,并刻画其结构特征

    这是设计动态规划算法的第一步,我们可以将矩阵连乘积AiAi+1……Aj记为A[i:j]。问题就是计算A[1:n]的最优计算次序。设这个计算次序在矩阵Ak和Ak+1之间将矩阵链断开,1<=k<n,使其完全加括号方式为((A1……Ak)(AK+1……An)),这样就将原问题分解为两个子问题,,按此计算次序,计算A[1:n]的计算量就等于计算A[1:k]的计算量加上A[k+1:n]的计算量,再加上A[1:k]和A[k+1:n]相乘的计算量。计算A[1:n]的最优次序包含了计算A[1:k]和A[k+1:n]这两个子问题的最优计算次序,以此类推,将A[1:k]和A[k+1:n]递归的分解下去,求出每个子问题的最优解,子问题的最优解相乘便得到原问题的最优解。

   (2).递归地定义最优值

           这是动态规划的第二步,对于矩阵连乘积的最优计算次序的问题,设计算A[i:j],1<=i<=j<=n,所需要的最小数乘次数为m[i][j],则原问题的最优值为m[1][n]。

           当i=j时,A[i:j]=Ai为单一的矩阵,则无需计算,所以m[i][j]=0,i=j=1,2,……,n。即对应的二维表对角线上的值全为0。

           当i<j时,这就需要用步骤(1)的最优子结构性质来计算m[i][j]。若计算A[i:j]的最优次序在Ak和Ak+1之间断开,i<=k<j,则m[i][j]=m[i][k]+m[k+1][j]+pi-1*pk*pj,k的位置只有j-i种可能,即k属于集合{i,i+1,……,j-1},所以k是这j-i个位置中使计算量达到最小的那个位置。

          所以m[i][j]可以递归地定义为        m[i][j]={  0                                                            i=j

                                                                      min{m[i][k]+m[k+1][j]+pi-1*pk*pj }         i<j ,i<=k<j     }

         将对应于m[i][j]的断开位置k记为s[i][j],在计算出最优值m[i][j]后,可递归地由s[i][j]构造出相应的最优解

   (3).以自底向上的方式计算出最优值

          动态规划的一大好处是,在计算的过程中,将已解决的子问题答案保存起来,每个子问题只计算一次,而后面的子问题需要用到前面已经解决的子问题,就可以从表中简单差出来,从而避免了大量的重复计算

         动态规划算法 这里的p[],m[][],s[][]都为全局变量

 1 void matrixChain(){
 2     for(int i=1;i<=n;i++)m[i][i]=0;
 3 
 4     for(int r=2;r<=n;r++)//对角线循环
 5         for(int i=1;i<=n-r+1;i++){//行循环
 6             int j = r+i-1;//列的控制
 7             //找m[i][j]的最小值,先初始化一下,令k=i
 8             m[i][j]=m[i][i]+m[i+1][j]+p[i-1]*p[i]*p[j];
 9             s[i][j]=i;
10             //k从i+1到j-1循环找m[i][j]的最小值
11             for(int k = i+1;k<j;k++){
12                 int temp=m[i][k]+m[k+1][j]+p[i-1]*p[k]*p[j];
13                 if(temp<m[i][j]){
14                     m[i][j]=temp;
15                     //s[][]用来记录在子序列i-j段中,在k位置处
16                         断开能得到最优解
17                     s[i][j]=k;
18                 }
19             }
20         }
21 }

        以A1A2A3A4A5A6为例,其中各矩阵的维数分别为:

      A1:30*35,  A2:35*15,  A3:15*5,  A4:5*10,  A5:10*20,  A6:20*25

      动态规划算法matrixchain计算m[ i ][ j ]先后次序如图所示,计算结果为m[ i ][ j ]和s[ i ][ j ],其中第0行和第0列没有使用。

     技术分享技术分享技术分享

            计算次序                                                      m[i][j]                                                             s[i][j]

     

              例如,在计算m[2][5]时,依递归式有

                  技术分享

               所以m[2][5] = 7125,且k=3,因此,s[2][5]=3。

   (4).根据计算最优值时得到的信息(及存放最优值的表格),构造最优解

         动态规划算法的第四布是构造问题的最优解。算法matrixchain只是计算出了最优值,并未给出最优解。也就是说,通过matrixchain的计算,只是到最少数乘次数,还不知道具体应按什么次序来做矩阵乘法才能达到最少的数乘次数。

        下面一段小程序按matrixchain计算出的断点矩阵s指示的加括号方式输出计算A[i:j]的最优计算次序

       

void printmatrix(int leftindex,int rightindex)//递归打印输出  
{  
    if(leftindex==rightindex)  
        printf("A%d",leftindex);  
    else{  
        printf("(");  
        printmatrix(leftindex,leftindex+s[leftindex][rightindex]);  
        printmatrix(leftindex+s[leftindex][rightindex]+1,rightindex);  
        printf(")");  
    }  

  到此,矩阵连乘问题就解决完了。

 四.源代码展示及运算结果

#include <stdio.h>  
#include <stdlib.h>  
#include <limits.h>  
#define MAX 50
int p[MAX+1];             //存储各个矩阵的列数以及第一个矩阵的行数(作为第0个矩阵的列数) 
int m[MAX][MAX];          //m[i][j]存储子问题的最优解  
int s[MAX][MAX];           //s[i][j]存储子问题的最佳分割点
int n;                    //矩阵个数
 
 void matrix(int n,int m[][n],int s[][n],int p[])  
{  
   
    int i,j,k;  
    for(i=0;i<n;i++)  
        m[i][i]=0;                  //最小子问题仅含有一个矩阵 ,对角线全为0 
                               
    for(i=2;i<=n;i++)  
        for(j=0;j<n-i+1;j++)
        {                         
            m[j][j+i-1]=INT_MAX;   
            for(k=0;k<i-1;k++)
            {                                    //k代表分割点  
                if(m[j][j+i-1]>m[j][j+k]+m[j+k+1][j+i-1]+p[j]*p[j+k+1]*p[j+i])
                {  
                    m[j][j+i-1]=m[j][j+k]+m[j+k+1][j+i-1]+p[j]*p[j+k+1]*p[j+i];  
                    s[j][j+i-1]=k;                           //记录分割点  
                }  
            }  
        }  
}  
  
 void printmatrix(int leftindex,int rightindex)//递归打印输出  
{  
    if(leftindex==rightindex)  
        printf("A%d",leftindex);  
    else{  
        printf("(");  
        printmatrix(leftindex,leftindex+s[leftindex][rightindex]);  
        printmatrix(leftindex+s[leftindex][rightindex]+1,rightindex);  
        printf(")");  
    }  
}  
int main()  
{  
    int i;
    printf("请输入矩阵相乘的矩阵个数");
    scanf("%d",&n);
    printf("请依次输入矩阵的行和烈(如A*B,A=20*30,B=30*40,即输入20 30 40)\n") ;
    for(i=0;i<n+1;i++)
    {
        scanf("%d",&p[i]);
    }
    matrix(n,m,s,p);  
    printf("矩阵连乘最小次数\t%d\n",m[0][n-1]);  
    printmatrix(0,n-1);  
    printf("\n");  
    return 0;  
}  

运行结果

技术分享

 

 

    

 

算法之矩阵连乘