题目描述
矩阵乘法的运算量与矩阵乘法的顺序强相关。
例如:
A是一个50×10的矩阵,B是10×20的矩阵,C是20×5的矩阵
计算A*B*C有两种顺序:((AB)C)或者(A(BC)),前者需要计算15000次乘法,后者只需要3500次。
编写程序计算不同的计算顺序需要进行的乘法次数
输入描述:
输入多行,先输入要计算乘法的矩阵个数n,每个矩阵的行数,列数,总共2n的数,最后输入要计算的法则
输出描述:
输出需要进行的乘法次数
示例1
输入
3 50 10 10 20 20 5 (A(BC))
输出
3500
代码:借用
//第六十八题 矩阵乘法计算量估算
#include<string>
#include<vector>
#include<iostream>
#include<stack>
using namespace std;
int main()
{
int N;
while (cin >> N)
{
vector<vector<int>> V(N, vector<int>(2, 0));//初始化二维数组
string rule;
for (auto &i : V)
for (auto &j : i)
cin >> j;
cin >> rule;
int count = 0;
stack<char> cal;//存储表达式的字母
for (int i = 0; i < rule.size(); i++)
{
if (rule[i] == ')')
if (cal.size() != 1)
{
vector<int> temp2 = V[cal.top() - 'A'];
cal.pop();
vector<int> &temp1 = V[cal.top() - 'A'];
count += temp1[1] * temp1[0] * temp2[1];
temp1[1] = temp2[1];
}
else;
else if (rule[i] != '(')
cal.push(rule[i]);
}
cout << count << endl;
}
return 0;
}