Matrix Chain Multiplication UVA——442

也可以看这里

题目链接

题目大意：给出若干个矩阵及其维度，再给出若干个表达式，计算每个表达式中矩阵相乘的次数，如果两个矩阵不能相乘则输出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;
}