藍橋杯演算法提高 矩陣乘法
阿新 • • 發佈:2019-01-04
原題:
演算法提高 矩陣乘法 時間限制: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 1005 long 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; }