POJ2246 Matrix Chain Multiplication (DFS)

题意分析

给出几个矩阵的规模，和几个矩阵的相乘的式子，计算这个矩阵相乘的式子是否合理，如果合理请给出乘法要计算多少次。

在处理矩阵表达式的时候有点类似 括号匹配的问题。 如果当前位置是一个左括号，那么我们就继续递归的调用处理表达式的函数，直到读到不是括号（即矩阵），然后就载入当前矩阵的行列信息。并返回该矩阵对象。

在读到两个矩阵之后，判断是否满足相乘的条件，即前一个矩阵的列数是否等于第二个矩阵的行数。若是则计算相乘的次数，$times= matrix1.row * matrix1.col * matrix2.col$
因为相乘最后结果矩阵行为 matrix1.row,列为 matrix2.col，共有$matrix1.row * matrix2.col$个元素，每个元素要称 $matrix1.col$从（或$matrix2.row$次）。

根据递归读取，依次调用，最后返回的matrix乘法次数即为结果。

代码总览

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef struct{
    int multime;
    int row;
    int col;
} matrix;
const int nmax = 1000;
char str[nmax];
int row[nmax],col[nmax];
bool iserror = false;
int n;
int point = 0;
matrix expression(){
    matrix t;
    if(iserror) return matrix{0,0,0};
    if(str[point] == '('){
        point++;
        matrix t1 = expression();
        matrix t2 = expression();
        point++;
        if(t1.col != t2.row){
            iserror = true;
            return matrix{0,0,0};
        }else{
            t.row = t1.row;
            t.col = t2.col;
            t.multime = t1.multime + t2.multime + t1.row * t2.col * t1.col;
        }
    }else{
        t.row = row[str[point] - 'A'];
        t.col = col[str[point] - 'A'];
        t.multime = 0;
        point++;
    }
    return t;
}
int main(){
    scanf("%d",&n);
    getchar();
    for(int i = 0;i<n;++i){
        char tmp;
        int t1,t2;
        scanf(" %c %d %d",&tmp,&t1,&t2);
        row[tmp-'A'] = t1;
        col[tmp-'A'] = t2;
    }
    while(scanf("%s",str) != EOF){
        iserror = false;
        point = 0;
        matrix ans = expression();
        if(iserror){
            printf("error\n");
        }else{
            printf("%d\n",ans.multime);
        }
        memset(str,0,sizeof str);
    }
    return 0;
}