矩阵连乘问题 动态规划

问题描述

给定n个矩阵，保证可以按顺序从第一个矩阵乘到第n个矩阵，你不可以改变矩阵的顺序，但是你可以在任意位置加入合法的括号，比如1234表示矩阵1×矩阵2×矩阵3×矩阵4，你可以加入括号1(23)4,1(234),来使得乘的顺序发生一定改变，要求求给定n个矩阵连乘其中乘法运算次数最少是多少。两个3×3的矩阵相乘需要27次乘法运算。

思路

首先看两个矩阵的乘法A×B,假设A的行数为r1列数为l1，B的行数为r2列数为l2，前提条件是l1必须等于r2,也就是第一个矩阵的列数必须等于第二个矩阵的行数，而且A×B的乘法运算次数为r1×l1×l2.并且得到矩阵行数是r1列数是l2.
因为题目保证n个矩阵一定可以乘到底，那么对于第i个矩阵，如果他小于n那么第i个矩阵的列数一定等于第i+1个矩阵的行数,用一个p数组去存储矩阵的信息，其中p[i-1]表示第i个矩阵的行数，p[i]表示第i个矩阵的列数。
先考虑递归求解这个问题，对于dfs(1,n)表示递归求解第1个矩阵乘到第n个矩阵，可以有如下代码

for(int i=2;i<n;i++)
dfs(1,n)=min(dfs(1,n),dfs(1,i)+dfs(i+1,n)+p[i-1]*p[i]*p[j];
//dfs(1,i)+dfs(i+1,n)+p[i-1]*p[i]*p[j] 相当于把第1~i的矩阵先乘了再把i+1到n先乘了然后两部分再相乘

进一步得出完整的dfs代码

ll dfs(int x,int y)
{
	if(vis[x][y])
	return vis[x][y];//vis[x][y]记录之后如果以后碰到直接输出

	if(x+1==y)
	return vis[x][y]=p[x-1]*p[x]*p[y];
	

	if(x==y)
	return 0;

	ll t=dfs(x,x)+dfs(x+1,y)+p[x-1]*p[x]*p[y];
	for(int j=x+1;j<y;j++)
		t=min(t,dfs(x,j)+dfs(j+1,y)+p[x-1]*p[j]*p[y]);
	return vis[x][y]=t;
}

在引入两个数组，一个数组dp[i] [j],dp[i] [j] 的含义为第i个矩阵乘到第j个矩阵所做乘法运算的最少次数。一个数组s[i] [j]，表示第i个矩阵乘到第j个矩阵中，所添加的最左边的右括号所在的位置。比如s[1][2] = 2,说明前第二个数组后面有个右括号。上述dfs自顶向下，我们采用自顶向上可以写出如下代码。

for(int i=1;i<=n;i++)
	dp[i][i]=0;

	for(int r=2;r<=n;r++)//枚举区间长度r
	{
		for(int i=1;i+r-1<=n;i++)//枚举起始位置
		{
			int j=i+r-1;//j为区间末尾位置
			dp[i][j]=dp[i][i]+dp[i+1][j]+p[i-1]*p[i]*p[j];
			s[i][j]=i;
			for(int k=i+1;k<j;k++)
			{
				ll t=dp[i][k]+dp[k+1][j]+p[i-1]*p[k]*p[j];
				if(t<dp[i][j])
				{
					dp[i][j]=t;
					s[i][j]=k;
				}
			}
		}
	}

完整代码如下

//动态规划
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
ll dp[1000][1000];//dp[i][j]表示第i个矩阵到第j个矩阵乘积最小值
ll p[1000];
int s[1000][1000];//s[i][j]表示第i个矩阵到第j个矩阵乘积最小值中 最左边的右括号位置
void solve()
{
	for(int i=1;i<=n;i++)
	dp[i][i]=0;

	for(int r=2;r<=n;r++)//枚举区间长度r
	{
		for(int i=1;i+r-1<=n;i++)//枚举起始位置
		{
			int j=i+r-1;//j为区间末尾位置
			dp[i][j]=dp[i][i]+dp[i+1][j]+p[i-1]*p[i]*p[j];
			s[i][j]=i;
			for(int k=i+1;k<j;k++)
			{
				ll t=dp[i][k]+dp[k+1][j]+p[i-1]*p[k]*p[j];
				if(t<dp[i][j])
				{
					dp[i][j]=t;
					s[i][j]=k;
				}
			}
		}
	}
}
void getans(int x,int y)//求矩阵乘积的次序
{
	if(x==y)
	{
		printf("%d",x);
		return;
	}
	printf("(");
	getans(x,s[x][y]);
	getans(s[x][y]+1,y);
	printf(")");
}
int main()
{
	scanf("%d",&n);//矩阵数量
	for(int i=1;i<=n;i++)
	{
		scanf("%lld%lld",&p[i-1],&p[i]);//每个矩阵的行数p[i-1]和列数p[i]
	}
	solve();
	printf("%lld\n",dp[1][n]);
	getans(1,n);
return 0;
}

//记忆化搜索
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n;
ll p[1000];
ll vis[1000][1000];
ll dfs(int x,int y)
{
	if(vis[x][y])
	return vis[x][y];

	if(x+1==y)
	return vis[x][y]=p[x-1]*p[x]*p[y];
	

	if(x==y)
	return 0;

	ll t=dfs(x,x)+dfs(x+1,y)+p[x-1]*p[x]*p[y];
	for(int j=x+1;j<y;j++)
		t=min(t,dfs(x,j)+dfs(j+1,y)+p[x-1]*p[j]*p[y]);
	return vis[x][y]=t;
}

int main()
{
	scanf("%d",&n);//矩阵数量
	for(int i=1;i<=n;i++)
	{
		scanf("%lld%lld",&p[i-1],&p[i]);//每个矩阵的行数p[i-1]和列数p[i]
	}

	cout<<dfs(1,n)<<endl;
	
return 0;
}