动态规划之矩阵连乘
题目:n个矩阵连乘,求最少的乘法运算次数以及结合方式
假设矩阵A为r1*r2,矩阵B为r2*r3,所以M=A*B=r1*r2*r3。当有多个矩阵相乘的时候,矩阵以不同的方式结合的时候其运算次数是不同的。
例如:M=M1 * M2 * M3 * M4
[5*20] [20*50] [50*1] [1*100]
((M1*M2)*M3)*M4=5000+250+500=5750
而M1*(M2*(M3*M4))=5000+100000+10000=115000
如果按照(M1*(M2*M3)*M4)=1000+100+500=1600
上面的例子说明根据不同的结合顺序,运算的次数是不同的。现在我们就是要求通过哪种顺序所进行的乘法运算次数最少,并且求出运算最少的次数是多少。
算法思想:
记Mi*Mi+1*Mi+2*...*Mj为mij,矩阵的大小为:M1为r1*r2,M2为r2*r3,M3为r3*r3,M4为r4*r5。r1,r2,r3,r4,r5分别为5,20,50,1,100.
(1)首先计算矩阵相乘个数为2的3种情况。
m12=M1*M2=r1*r2*r3=5000,m23=r2*rr*r4,m34=r3*r4*r5=1000,m34=5000
(2)计算相乘矩阵个数为3的2种情况
m13=M1*M2*M3,目标是求出m13的乘法运算次数最小值。m13=min{M1*m23+r1*r2*r4,m12*M3+r1*r3*r4}=min{1100,5250}=m12*M3+r1*r3*r4=1100
同理 m24=M2*M3*M4=min{M2*m34+r2*r3*r,m23*M4+r2*r4*r5}=min{10500,3000}=3000;
(3)最后计算相乘矩阵个数为4的最终结果:
m14=min(m13*M4+r1*r4*r5,m12*m34+r1*r3*r4,M1*m24+r1*r2*r5)={3000+10000 ,5000+5000,+25000,1100+500}=1600
算法如下
1 #include <iostream> 2 3 using namespace std; 4 5 int r[100],com[100][100]; 6 7 int course(int i,int j) 8 { 9 int u,t; 10 if(i==j) 11 return 0; 12 if(i==j-1) 13 { 14 com[i][j]=i; 15 return r[i]*r[i+1]*r[i+2]; 16 } 17 else 18 { 19 u=course(i,i)+course(i+1,j)+r[i]*r[i+1]*r[j+1]; 20 com[i][j]=i; 21 for(int k=i;k<j;k++) 22 { 23 t=course(i,k)+course(k+1,j)+r[i]*r[k+1]*r[j+1]; 24 if(t<u) 25 { 26 u=t; 27 com[i][j]=k; 28 } 29 } 30 return u; 31 } 32 } 33 34 35 int main() 36 { 37 int n; 38 printf("input n:"); 39 scanf("%d",&n); 40 for(int i=1;i<=n;i++) 41 scanf("%d",&r[i]); 42 printf("min=%d\n",course(1,n-1)); 43 for(int i=1;i<=n;i++) 44 { 45 for(int j=1;j<=n;j++) 46 printf("%d ",com[i][j]); 47 printf("\n"); 48 } 49 return 0; 50 }
com适用于保存矩阵的结合方式的com[i][j]=k,表示的是mik*mkj的时候进行乘法的次数最少
递归调用的过程如下:
1-2,2-3,3-4等子问题都被递归调用了2次,子问题存在重复。可以采用备忘录方法解决该问题,设置一个全局变量m[i][j[,存储已经计算过的course(i,j)的值,下次调用的时候直接使用。