区间dp 矩阵连乘

题目描述

给定n个矩阵{A1,A2,...,An}，及m个矩阵连乘的表达式，判断每个矩阵连乘表达式是否满足矩阵乘法法则，如果满足，则计算矩阵的最小连乘次数，如果不满足输出“MengMengDa“。

输入

输入数据由多组数据组成（不超过10组样例）。每组数据格式如下：
第一行是2个整数n (1≤n≤26)和m(1≤m≤3)，表示矩阵的个数。
接下来n行，每行有一个大写字母，表示矩阵的名字，后面有两个整数r和c，分别表示该矩阵的行数和列数，其中1<r, c<100。
第n+1行到第n+m行，每行是一个矩阵连乘的表达式(2<=矩阵个数<=100)。

输出

对于每个矩阵连乘表达式，如果运算不满足矩阵乘法法则的情况（即左矩阵列数与右矩阵的行数不同），则输出“MengMengDa”，否则输出最小矩阵连乘次数。

数据保证结果不超过1e9。

 

样例输入

A 10 100
B 5 50
C 100 5
ACB
ABC

样例输出

7500
MengMengDa

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;

struct mm{
    int r;
    int c;
}ans[30];


string s;
int f[105][105];

void pre(){
    memset(f,0x3f,sizeof(f));
    for (int i = 0; i < 105; ++i)
         f[i][i] = 0;
}

int dp(){
    for (int j = 0;j < s.size() - 1; ++j)
        if (ans[s[j] - 'A'].c != ans[s[j + 1] - 'A'].r) return -1;
    for (int l = 2; l <= s.size(); ++l)
        for (int i = 0; i+l-1 < s.size(); ++i){
            int j = i+l-1;
            for (int k = i; k < j; ++k){
                f[i][j] = min(f[i][j],f[i][k] + f[k+1][j] + ans[s[i] - 'A'].r * ans[s[k] - 'A'].c * ans[s[j] - 'A'].c); 
            }
        }
    return f[0][s.size() - 1];
}

int main(){
    int n,m,x,y;
    char matrix;
    while(~scanf("%d%d",&n,&m)){
        while(n--){
            cin>>matrix;
            scanf("%d%d",&x,&y);
            ans[matrix - 'A'].r = x;
            ans[matrix - 'A'].c = y;
        }
        
        while(m--){
            cin>>s;
            pre();
            int t = dp();
            if (t == -1||t==1e18) puts("MengMengDa");
            else printf("%d\n",t);
        }
    }
    return 0;
}