蓝桥杯算法提高 矩阵乘法

来源:互联网 发布:网络时间在线 编辑:程序博客网 时间:2024/06/05 20:45

原题:

  算法提高 矩阵乘法  
时间限制:3.0s   内存限制:256.0MB
    
问题描述
  有n个矩阵,大小分别为a0*a1, a1*a2, a2*a3, ..., a[n-1]*a[n],现要将它们依次相乘,只能使用结合率,求最少需要多少次运算。
  两个大小分别为p*q和q*r的矩阵相乘时的运算次数计为p*q*r。
输入格式
  输入的第一行包含一个整数n,表示矩阵的个数。
  第二行包含n+1个数,表示给定的矩阵。
输出格式
  输出一个整数,表示最少的运算次数。
样例输入
3
1 10 5 20
样例输出
150
数据规模和约定
  1<=n<=1000, 1<=ai<=10000。
分析:

和上次的合并石子一样的动态规划,只不过是变为乘法在相加。

代码:

#include <iostream>#include <string.h>using namespace std;#define maxn 0x3f3f3f3f#define N 1005long long  dp[N][N],p[N][N],sum[N];int main(){    int n,x;    cin>>n;    memset(dp, 1, sizeof(dp));    for(int i = 0;i <= n;i++)    {        cin>>sum[i];        dp[i][i] = 0;    }    for(int l = 2;l <= n;l++)    {        for(int i = 1;i <= n - l + 1;i++)        {            long long j = i + l - 1,temp = dp[i + 1][j] + sum[i - 1] * sum[i] * sum[j];            for(int k = i + 1;k < j;k++)            {                long long t = dp[i][k] + dp[k + 1][j] + sum[i - 1] * sum[k] * sum[j];                temp = min(temp,t);            }            dp[i][j] = temp;        }    }    cout<<dp[1][n];    return 0;}


1 0
原创粉丝点击