题目大意:给出若干个矩阵及其维度,再给出若干个表达式,计算每个表达式中矩阵相乘的次数,如果两个矩阵不能相乘则输出error,表达式中有括号,但是每个括号中只有两个矩阵。
思路:栈的应用
这个题特殊在了每个括号中只有两个矩阵这个条件上,所以我们可以不用储存左括号,一旦遇到右括号,直接弹出两个判断即可。
代码:
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 30 , M = 100010;
typedef pair < int , int > PII;
int n;
PII matr[N];//矩阵数组
PII stk[M];//数组模拟栈
char str[M];//表达式字符串
int solve()//计算整个表达式,如果有非法情况则返回-1
{
int top = 0;//栈数数组的下标
int res = 0;//表达式结果
for(int i = 0 ; str[i] ; ++ i)
{
//如果是一个字母,直接将维度压入栈中
if(isalpha(str[i])) stk[top ++] = matr[str[i] - 'A'];
//如果是右括号,直接弹出两个判断即可
else if(str[i] == ')')
{
PII b = stk[-- top];
PII a = stk[-- top];
if(a.second != b.first) return -1;
else
{
res += a.first * a.second * b.second;
stk[top ++] = {a.first , b.second};
}
}
}
return res;
}
int main(void)
{
scanf("%d",&n);
for(int i = 0 ; i < n ; ++ i)//读入数据
{
char n[2];
int a , b;
scanf("%s %d %d",n , &a , &b);
matr[n[0] - 'A'] = {a , b};
}
while(scanf("%s",str) != EOF)
{
int res = solve();
if(res == -1) puts("error");//发生了错误
else printf("%d\n",res);
}
return 0;
}
其实上面那个是dalao的思路,我自己在做的时候是看的紫书的翻译,所以不知道每个括号中只有两个矩阵的,emmm,但其实也就是简单的括号匹配问题。
代码:
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 30 , M = 100010;
typedef pair < int , int > PII;
int n , top;
PII stk[M];
char str[M];
int row[N] , col[N];
int judge (PII&a , PII&b)
{
int r1 = a.first , c1 = a.second;
int r2 = b.first , c2 = b.second;
if(r1 > 0 && c1 > 0 && r2 > 0 && c2 > 0 && r2 == c1) return r1 * c1 * c2;
else return -1;
}
int main(void)
{
scanf("%d",&n);
for(int i = 0 ; i < n ; ++ i)//读入数据
{
char n[2];
int a , b;
scanf("%s %d %d",n , &a , &b);
int idx = n[0] - 'A';
row[idx] = a;
col[idx] =b;
}
while(scanf("%s",str) != EOF)
{
top = 0;
if(strlen(str) == 1)//长度是1的话直接打印0即可
{
puts("0");
continue;
}
int res = 0;
for(int i = 0 ; str[i] ; ++ i)
{
if(str[i] == '(') stk[top ++] = {0 , 0};//左括号以一个特殊的维度压入栈中
else if(str[i] != ')') stk[top ++] = {row[str[i] - 'A'] , col[str[i] - 'A']};
else//右括号,循环出站判断
{
while(top && stk[top - 2] != (PII){0 , 0})
{
//弹出两个
PII b = stk[-- top];
PII a = stk[-- top];
if(a == (PII){0 , 0} || b == (PII){0 , 0}) break;
int calc = judge(a , b);
if(calc == -1)//结果是非法
{
res = -1;
break;
}
else//结果合法
{
int r1 = a.first , c1 = a.second;
int r2 = b.first , c2 = b.second;
res += calc;
if(top > 1 && stk[top - 1] == (PII){0 , 0}) stk[top - 1] = {r1 , c2};
else stk[top ++] = {r1 , c2};
}
}
}
if(res == -1) break;
}
if(res == -1) puts("error");
else printf("%d\n",res);
}
return 0;
}