题目:
以单个大写字母代替矩阵,第一行输入矩阵种类的数量,后输入各矩阵代替字母及维度(行数 列数)
然后输入多行矩阵表达式,输出每个表达式乘法计算次数
注:1.每对括号内有且必有两个矩阵,如果有套娃现象 例如:(A(BC))将 (BC) 看作一个矩阵
2.乘法不合规时输出 “error” ,即不满足 “前矩阵的列数 = 后矩阵的行数”
Sample Input
9
A 50 10
B 10 20
C 20 5
D 30 35
E 35 15
F 15 5
G 5 10
H 10 20
I 20 25
A
B
C
(AA)
(AB)
(AC)
(A(BC))
((AB)C)
(((((DE)F)G)H)I)
(D(E(F(G(HI)))))
((D(EF))((GH)I))
Sample Output
0
0
0
error
10000
error
3500
15000
40500
47500
15125
https://vjudge.net/problem/UVA-442
思路:
参考紫书:利用栈解析表达式
忽略 ‘(’ ,遇到大写字母存入栈中,遇到 ‘)’ 从栈顶取出两个矩阵计算乘法次数,然后累加,将新矩阵(r1,c2)压栈,重复
代码:
结合思路有如下两种实现
pair+stack实现:
#include <iostream>
#include <string>
#include <stack>
#include <map>
using namespace std;
typedef pair<int, int> pairr;
map<char, pairr> mp;
pairr ryc, r1, r2;
int main() {
int T; cin >> T;
int a, b; char ch;
while (T--) {
cin >> ch >> a >> b;
mp[ch] = make_pair(a, b);//记录矩阵
} string t;
while (cin >> t) {
int ans = 0;
stack<pairr> s;
bool error = false;
for (int i = 0; i < t.length(); i++) {
if (isalpha(t[i]))
s.push( pairr( mp[t[i]] ) );//矩阵直接压栈
if (t[i] == ')') {
r2 = s.top(); s.pop();//后一个
r1 = s.top(); s.pop();//前一个
if (r1.second != r2.first) {
error = true; break;
} else {
ans += r1.first * r1.second * r2.second;
s.push(pairr(r1.first, r2.second));//压栈
}
}
}
if (error) cout << "error" << endl;
else cout << ans << endl;
}
return 0;
}
struct+stack实现:
#include <cstdio>
#include <iostream>
#include <stack>
using namespace std;
struct Matrix {
int r,c;
Matrix(int r=0, int c=0):r(r),c(c) {};
}a[26];
stack<Matrix> mar;/* used to put Matrix */
int main() {
char ch;
int n;cin >> n;
for(int i=0;i<n;i++) {
getchar();cin >> ch;
cin >> a[ch-'A'].r >> a[ch-'A'].c;
} string s;
while(cin >> s) {
int sum=0;int error = 0;
for(int i=0;i<s.length();i++)/* */ {
if(isalpha(s[i])) mar.push(a[ s[i]-'A' ]);
if(s[i]==')') {
Matrix a2 = mar.top(); mar.pop();/* modify a1*a2 */
Matrix a1 = mar.top(); mar.pop();
if(a1.c != a2.r) { error = 1;break; }
sum+=a1.r*a1.c*a2.c;
mar.push(Matrix(a1.r,a2.c));/* New Matrix join_into */
}
}
if(error) cout << "error" << endl;
else cout << sum << endl;
}
return 0;
}
两种都可以实现,只是载体不同 ( •̀ .̫ •́ )✧