矩阵链乘 Matrix Chain Multiplication 洛谷

题目链接

题目：

# 矩阵链乘 Matrix Chain Multiplication

## 题面翻译

## 矩阵链乘

### 题目描述

​  假设你必须评估一种表达形如 A*B*C*D*E，其中 A,B,C,D,E是矩阵。既然矩阵乘法是关联的，那么乘法的顺序是任意的。然而，链乘的元素数量必须由你选择的赋值顺序决定。

​  例如，A，B，C分别是 50 * 10 ，10 * 20 和 20 * 5 的矩阵。现在有两种方案计算 A * B * C ,即（A * B) * C 和 A*(B * C)。  
   第一个要进行15000次基本乘法，而第二个只进行3500次。

​  你的任务就是写出一个程序判定以给定的方式相乘需要多少次基本乘法计算。

### 输入格式

​  输入包含两个部分：矩阵和表达式。 
   输入文件的第一行包含了一个整数 n(1  $\leq$  n  $\leq$  26), 代表矩阵的个数。接下来的n行每一行都包含了一个大写字母，说明矩阵的名称，以及两个整数，说明行与列的个数。  
   第二个部分严格遵守以下的语法：

SecondPart = Line {  Line  } <EOF>
Line       = Expression <CR>
Expression = Matrix | "(" Expression Expression ")"
Matrix     = "A" | "B" | "C" | ... | "X" | "Y" | "Z"

###输出格式

​  对于每一个表达式，如果乘法无法进行就输出 " error "。否则就输出一行包含计算所需的乘法次数。 

感谢@onceagain 提供翻译

## 题目描述

[problemUrl]: https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=6&page=show_problem&problem=383

[PDF](https://uva.onlinejudge.org/external/4/p442.pdf)


## 输入格式


## 输出格式


## 样例 #1

### 样例输入 #1

```
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))
```

### 样例输出 #1

```
0
0
0
error
10000
error
3500
15000
40500
47500
15125
```

思路：

1.使用栈解决；

2.首先读取字符串

        当遇到 '(' 符号时，直接继续读取下一个字符 （提速）

        当遇到 字母 时，存入栈中

        当遇到 ')' 符号时，连续弹出栈顶两次，开始进行判断与计算

3.判断两个矩阵是否能进行运算，即

        矩阵a的n列是否与矩阵b的m行相等，若相等，进行计算；若不相等，直接终止，打印"error"

4.计算两个矩阵 a 与 b 的运算次数为矩阵 a.m * a.n * b.n

5.每计算两个矩阵运算次数，就将两个矩阵合成一个新矩阵入栈

6.提速方面，除了步骤2中的一点，还有：

        如果字符只有一个，那么答案直接为0，因为一个矩阵不需要进行任何运算

完整代码如下：

#include<iostream>
#include<stack>
#include<cstdlib>

using namespace std;

struct Matrix
{
	int m, n;
};

string qustion;

long long sum;

int check ( Matrix x, Matrix y )
{
	if ( x.n != y.m )
	{
		return 0;
	}
	return 1;
}

void count( Matrix x, Matrix y, Matrix *z )
{
	sum += x.m * x.n * y.n;
	z->m = x.m;
	z->n = y.n;
}

void solve ( Matrix *matrix )
{
	stack<Matrix> help;
	sum = 0;
	int length = qustion.length();
	int i;
	Matrix x, y, z;
	int flag = 1;
	for ( i = 0; i < length; ++ i )
	{
        if ( qustion[ i ] == '(' )
        {
            continue;
        }
		if ( qustion[ i ] != '(' && qustion[ i ] != ')' )
		{
			help.push(matrix[ qustion[ i ] - 'A' ] );
		}
		if ( qustion[ i ] == ')' )
		{
			y = help.top();
			help.pop();
			x = help.top();
			help.pop();
			
			if ( check ( x, y ) )
			{
				count ( x, y, &z );
				help.push( z );
			}
			else
			{
				flag = 0;
				break;
			}
		}
	}
	if ( flag )
	{
		printf ( "%lld\n", sum );
	}
	else
	{
		puts ( "error" );
	}
}

int main ( )
{
	Matrix *matrix;
	matrix = ( Matrix *)malloc( 26 * sizeof ( Matrix ) );
	int n;
	scanf ( "%d", &n );
	getchar();
	int i;
	char a;
	int b;
	for ( i = 0; i < n; ++ i )
	{
		scanf ( "%c", &a );
		b = a - 'A';
		scanf ( "%d %d", &matrix[ b ].m, &matrix[ b ].n );
		getchar();
	}
	while ( cin >> qustion )
	{
        if ( qustion.length() == 1 )
        {
            puts ( "0" );
        }
        else
        {
            solve(matrix);
        }
	}
	return 0;	
}