蓝桥杯 ADV-232 矩阵乘法

算法提高 矩阵乘法  

时间限制：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。

分析：分析一个序列2 7 5 1 4。首先三个三个的比较，乘积和分别是70,35,20。选择35，合并后序列为2 7 1 4。接下来乘积和分别是14和28，选择14。此时次数为35 + 14 = 49。最后序列为2 1 4，乘积和为8，次数为49 + 8 = 57。

我们建立二维的dp数组。用dp[i][j]来记录第i个到第j个元素之间所能选取到的最小的次数。当第i个元素到第j个元素仅有三个时（如 3 2 5），dp[i][j]为三者乘积（3 * 2 * 5）。若第i个元素到第j个元素有四个时（如 3 2 5 7），dp[0][2] = 3 * 2 * 5 = 30，dp[1][3] = 2 * 5 * 7 = 70。此时两种序列分别是3 5 7与3 2 7，这两种乘积分别是105和42。选择最小的和70 + 42 = 112。dp[0][3] = 112 = dp[0][1]（值为0） + dp[1][3] + a[0] * a[1] * a[3]。

要点在于，每次得到的三个值的乘积，设三项的下标分别是a, b, c。由于三项可能是由多项合并得到的，需要将此处的次数增加三项下标之间对应的次数（dp[a][b], dp[b][c]）。

这里可以得到递推公式。

还需要注意，数据用long long，INF也需要开的大一点，否则无法通过。

代码：

#include<iostream>
#include<algorithm>
#define INF 0x3ffffffffff
#define MAXN 1010
using namespace std;
int main() {
	int n;
	cin >> n;
	n++;
	long long a[MAXN] = { 0 };
	long long dp[MAXN][MAXN] = { 0 };
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	fill(dp[0], dp[0] + MAXN * MAXN, INF);
	for (int i = 0; i < n - 1; i++) {
		dp[i][i + 1] = 0;
	}
	for (int len = 3; len <= n; len++) {
		for (int i = 0; i < n - len + 1; i++) {
			int j = i + len - 1;
			for (int k = i + 1; k < j; k++) {
				dp[i][j] = min(dp[i][j], dp[i][k] + dp[k][j] + a[i] * a[k] * a[j]);
			}
		}
	}
	cout << dp[0][n - 1] << endl;
	return 0;
}