Matrix Chain Multiplication UVA - 442(栈 + 模拟 + 表达式解析)

题目:

以单个大写字母代替矩阵,第一行输入矩阵种类的数量,后输入各矩阵代替字母及维度(行数 列数)
然后输入多行矩阵表达式,输出每个表达式乘法计算次数
注: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;
}

两种都可以实现,只是载体不同 ( •̀ .̫ •́ )✧

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值