1. 程式人生 > >【藍橋】演算法提高 矩陣乘法

【藍橋】演算法提高 矩陣乘法

 演算法提高 矩陣乘法   時間限制: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<cstdio>
#include<math.h>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<climits>
#include<string>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<algorithm>
using namespace std;
#define rep(i,j,k)for(i=j;i<k;i++)
#define per(i,j,k)for(i=j;i>k;i--)
#define MS(x,y)memset(x,y,sizeof(x))
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define ll long long
#define abs(x) (x>0?x:-x)
const int INF=0x7ffffff;
const int SIZE=1000+10;
const ll MAX=1e18;

ll d[SIZE][SIZE];
ll m[SIZE];
int i,j,k,n;

ll matrix_DP(int n)
{
    ll sum;
    MS(d,0);
    for(int len=2;len<=n;len++){
        for(int i=1,j=len;j<=n;i++,j++){
            ll minx=MAX;
            for(int k=i;k<j;k++){
                sum=d[i][k]+d[k+1][j]+m[i-1]*m[k]*m[j];
                if(sum<minx)
                    minx=sum;
            }
            d[i][j]=minx;
        }
    }
    return d[1][n];
}

int main()
{
    while(~scanf("%d",&n)){
        for(i=0;i<=n;i++)
            scanf("%lld",&m[i]);
        printf("%lld\n",matrix_DP(n));
    }
    return 0;
}