码迷,mamicode.com
首页 > 其他好文 > 详细

31-矩阵相乘

时间:2017-09-28 20:43:09      阅读:186      评论:0      收藏:0      [点我收藏+]

标签:comm   区间dp   blog   思路   选择   ...   输出   决定   str   

                  C时间限制:3000 毫秒 |  C内存限制:3000 Kb
题目内容:
 n个矩阵A1,A2,...,An, ,如果其维度分别为d0xd1, d1xd2, ...dn-1xdn,则可以进行连乘运算A1A2A3..An . 
连乘运算可以采取不同的顺序进行,如(A1A2)A3 和 A1(A2A3),这两种顺序的消耗乘法计算量是不同的,前者是
d0.d1.d2+d0.d2.d3,后者d0.d1.d3+d1.d2.d3.

第1步选择哪两个,第2步又选择哪两个,一直到最后算出结果需要做n-1次决定。其中有一种计算顺序使得
A1A2A3..An总的乘法计算量最小。
输入描述
第一行输入n
第二行输入维度向量d0,d1,...dn

输出描述
输出所需的最少乘法次数。

输入样例
4
2 3 2 4 3

输出样例
48
思路: 区间dp, 设dp[i][j]表示第i个矩阵到第j个矩阵之间合并的最少相乘次数
则递归方程为: dp[i][j] = min(dp[i][j], dp[i][k] + dp[k + 1][j] + d[i-1]*d[k]*d[j]); i < k < j;
需要知道相邻矩阵的乘法规则:A1(x1, y1)* A2(x2, y2) = x1 * y1 * y2;(x表示行,y表示列)
#include <iostream>
using namespace std;
int dp[100][100];
int a[100];
int n;

int fun(int i, int j){
	if(i == j)
		return dp[i][j];
	if(i == j - 1)
		return dp[i][j] = a[i - 1] * a[i] * a[j];
	if(i < j - 1){
		dp[i][j] = 10000000;
		for(int k = i + 1; k < j; k++)
			dp[i][j] = min(dp[i][j], fun(i, k) + fun(k + 1, j) + a[i - 1] * a[j] * a[k]);
	}
	return dp[i][j];
}

int main(){
	cin >> n;
	
	for(int i = 0; i <= n; i++)
		cin >> a[i];
	cout << fun(1, n);
	
	return 0;
}

 

31-矩阵相乘

标签:comm   区间dp   blog   思路   选择   ...   输出   决定   str   

原文地址:http://www.cnblogs.com/zhumengdexiaobai/p/7608236.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!