ZOJ 1094_Matrix Chain Multiplication

大意：输入一系列矩阵（不超过26个），判断下面输入的矩阵乘法是否合法（即是否满足前者的列等于后者的行），若合法则输出其做了多次数的乘法运算（若有两矩阵A：a,b; B: b,c ,则其做的数乘法次数为a*b*c），若不合法，则输出error。

 

分析：为了操作方便定义一个矩阵结构，保存矩阵名，row和column. 从题意可知，每次的乘法都用括号隔开，那么这里就符合LIFO。前括号可以忽略，后括号作为每次乘法运算的标识。遇到矩阵即压栈。

 

#include<iostream>
#include<stack>
#include<string>
using namespace std;
struct Matrix//定义矩阵元素 
{
	char name;
	int row;
	int column;
};
int main()
{
	stack<Matrix> sk;
	int i,n,j,result;
	Matrix M[26],a,b,temp;
	string s;//保存每一组输入 
	bool flag;//error标识 
	while(cin>>n)
	{
		for(i=0;i<n;i++)
			cin>>M[i].name>>M[i].row>>M[i].column;
		while(cin>>s)
		{
			result=0;
			i=0;
			flag=true;
			while(s[i]!='\0')
			{
				if(s[i]!='(')//过滤'(' 
				{
					if(s[i]>='A'&&s[i]<='Z')//遇到字母即在已知的矩阵数组中匹配 
					{
						for(j=0;j<n;j++)
						{
							if(M[j].name==s[i])
							{
								sk.push(M[j]);//压栈 
								break;
							}
						}
					}
					else
					{
						if(s[i]==')')//')'为乘法标识 
						{
							b=sk.top();
							sk.pop();
							a=sk.top();
							sk.pop();
							if(a.column==b.row)//符合矩阵乘法法则 
							{
								temp.name='M';
								temp.row=a.row;
								temp.column=b.column;
								sk.push(temp);//将结果压栈 
								result+=a.row*a.column*b.column;//累计乘法次数 
							}
							else//不符合即error并标识不输出result 
							{
								cout<<"error"<<endl;
								flag=false;
								break;
							}
						}
					}
				}
				i++;
			}
			while(!sk.empty())//清空栈 
				sk.pop();
			if(flag)
				cout<<result<<endl;
		}
	}
	return 0;
}