【藍橋】演算法提高 矩陣乘法
阿新 • • 發佈:2018-12-30
演算法提高 矩陣乘法
時間限制: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。
兩個大小分別為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; }