【栈 && 矩阵的知识】UVA 442 Matrix Chain Multiplication

来源:互联网 发布:邮件软件中文字体 编辑:程序博客网 时间:2024/06/11 17:02

Problem Description

给你n个矩阵,每个矩阵由一个字母命名,后面跟着矩阵的n行m列。
输入字符串要你求它相乘的次数,如果不能相乘输出error

代码:A的列数不等于B的行数不能相乘,假定A是m*n矩阵, B是n*p矩阵,那么AB是m*p矩阵,乘法次数m*n*p..

#include<bits/stdc++.h>using namespace std;struct node{    int a, b;};node m[26];stack<node> q;//存对应的值int main(){    int n, k, i, len, ans;    while(cin >> n)    {        string name;        for(i = 0; i < n; i++)//先把对应字母的矩阵用结构体存起来        {            cin >> name;            k = name[0] - 'A';            cin >> m[k].a >> m[k].b;        }        string s;        node x1, x2;        while(cin >> s)        {            len = s.length();            ans = 0;            for(i = 0; i < len; i++)            {                if(s[i] == ')')//出来两个                {                    x2 = q.top(); q.pop();                    x1 = q.top(); q.pop();                    if(x1.b != x2.a) {printf("error\n"); break;}//不满足矩阵相乘                    else                    {                        ans += x1.a * x1.b * x2.b;//满足,求乘的次数                        q.push((node){x1.a, x2.b});//存起来                    }                }                else if(s[i] >= 'A' && s[i] <= 'Z')//存起来                q.push((node){m[s[i]-'A'].a, m[s[i]-'A'].b});            }            if(i == len) printf("%d\n", ans);            while(!q.empty()) q.pop();        }    }    return 0;}
阅读全文
0 0
原创粉丝点击