问题 Q: 数据结构:矩阵连乘问题

题目描述
给定n个矩阵{A1,A2,…,An},考察这n个矩阵的连乘积A1A2…An。由于矩阵乘法满足结合律,故计算矩阵的连乘积可以有许多不同的计算次序,这种计算次序可以用加括号的方式来确定。

矩阵连乘积的计算次序与其计算量有密切关系。例如,考察计算3个矩阵{A1,A2,A3}连乘积的例子。设这3个矩阵的维数分别为10100,1005,和550。若按(A1A2)A3计算,3个矩阵连乘积需要的数乘次数为101005+10550 = 7500。若按A1(A2A3)计算,则总共需要100550+10100*50 = 75000次数乘。

现在你的任务是对于一个确定的矩阵连乘方案,计算其需要的数乘次数。

输入
输入数据由多组数据组成。每组数据格式如下:
第一行是一个整数n (1≤n≤26),表示矩阵的个数。
接下来n行,每行有一个大写字母,表示矩阵的名字,后面有两个整数a,b,分别表示该矩阵的行数和列数,其中1<a,b<100。
第n+1行是一个矩阵连乘的表达式,由括号与大写字母组成,没有乘号与多余的空格。如果表达式中没有括号则按照从左到右的顺序计算,输入的括号保证能够配对。
输出
对于每组数据,输出仅一行包含一个整数,即将该矩阵连乘方案需要的数乘次数。如果运算过程中出现不满足矩阵乘法法则的情况(即左矩阵列数与右矩阵的行数不同),则输出“error”。

样例输入 Copy
3
A 10 100
B 100 5
C 5 50
A(BC)
样例输出 Copy
75000

#include<stack>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e4;
int T,n,tot=26,flag,ans;
char s[maxn];
 
struct rec
{
	int x,y;
	rec(int x=0,int y=0):x(x),y(y){}
	void read(){scanf("%d%d",&x,&y);}
}a[maxn];
 
rec operator*(rec a,rec b)
{
	if (a.y!=b.x) flag=0;
	ans+=a.x*a.y*b.y;
	return rec(a.x,b.y);
}
 
int main()
{
	//scanf("%d",&T);
	while (~scanf("%d",&n))
	{
		flag=1;	ans=0;
		for (int i=0;i<n;i++)
		{
			scanf("%s",s);
			a[s[0]-'A'].read();
		}
		scanf("%s",s);
		stack<int> p,q;
		for (int i=0;s[i];i++)
		{
			if (s[i]=='(')  p.push(-1);
			else if (s[i]==')')
			{
				while (p.top()!=-1) {q.push(p.top()); p.pop();}
				p.pop();	int k=q.top();	q.pop();
				while (!q.empty())
				{
					a[++tot]=a[k]*a[q.top()];
					k=tot;	q.pop();
				}
				p.push(k);
			}
			else p.push(s[i]-'A');
		}
		while (!p.empty()) {q.push(p.top()); p.pop(); }
		int k=q.top();	q.pop();
		while (!q.empty())
		{
			a[++tot]=a[k]*a[q.top()];
			k=tot;	q.pop();
		}
		if (flag) printf("%d\n",ans);
		else printf("error\n");
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值