注: 此题比较无脑, 且没有意思, 建议不要做.
题意:
矩阵乘法, 要求计算出几个矩阵相乘时需要进行的相乘次数. 输入分为 first part 与 second part, first part 给出各个矩阵的行数与列数, second part 格式如下:
SecondPart = Line { Line } <EOF>
Line = Expression <CR>
Expression = Matrix | "(" Expression Expression ")"
Matrix = "A" | "B" | "C" | ... | "X" | "Y" | "Z"
即每一行包含由括号括起来的表达式或矩阵, 按括号内的表达式先运算的规则进行矩阵乘法, 给出相乘的总次数.
思路:
1. 对于矩阵 A(m,n), 矩阵 B(n,p), 则 AB 的相乘次数为 m*n*p;
2. 题目是不需要考虑括号是否匹配的, 即 (AB)) 的结果不能是 error, 必须返回正常 AB 的结果, 所以按最无脑的写法, 碰到 字母 就进栈, 碰到 ) 就出栈两个数, 进行运算后入栈, 这样就可以 AC 了.
3. 使用 map 来存放每个矩阵的行与列.
4. 若 A 与 B 相乘, 则结果 AB 做为矩阵名, A 的行数做为 AB 的行, B 的列数做为 AB 的列, 并将结果存入 map 中.
5. 看到别人的代码直接使用一个结构体, 包含行与列, 即不考虑矩阵的名称, 直接把乘出来的结果生成一个新的结构体, 只包含行与列, 然后压栈, 这样可以不用使用 map, 也比较简单.
6. 本题有一个比较 SB 的地方就是, second part 输入有可能是空行, 此时, 必须不作处理. 一开始我写的代码, 会把空行的结果当 0 输出(即不做运算), 导致无限 WA; 题目这样太无聊了, 未说明的东西, 也放到 test case 里, 这不是拆腾人嘛.
要点:
char c 转 string: string str(1, c); 或 ""+c;
代码:
# include <iostream>
# include <string>
# include <cstdio>
# include <cstring>
# include <vector>
# include <algorithm>
# include <cctype>
# include <iterator>
# include <assert.h>
# include <map>
# include <stack>
using namespace std;
map< string, pair<int, int> > MATRIXS;
// 计算两个矩阵相乘需要的相乘元素个数, 若两个矩阵不能相乘, 则返回 -1
// 两矩阵可以相乘是指 left 的 column 要等于 right 的 row
// 两矩阵的总乘数是指 left.row * left.column * right.columnl
int getNumMulti(const string& left, const string& right) {
if (MATRIXS.find(left) == MATRIXS.end() ||
MATRIXS.find(right) == MATRIXS.end()) {
return -1;
}
pair<int, int> matrixsLeft = MATRIXS[left];
pair<int, int> matrixsRight = MATRIXS[right];
if (matrixsLeft.second == matrixsRight.first) {
return matrixsLeft.first * matrixsLeft.second * matrixsRight.second;
} else {
return -1;
}
}
// 计算需要进行相乘的次数, 返回 -1 表示出错了
int getNumMulti(const string& line) {
stack<string> stk;
int numMulti = 0;
for (int i=0; i<line.size(); i++) {
if (isupper(line[i])) {
string matrix(1, line[i]);
stk.push(matrix);
} else if (line[i] == ')') {
string right = stk.top();
stk.pop();
string left = stk.top();
stk.pop();
int num = getNumMulti(left, right);
if (num == -1) return -1;
numMulti += num;
string newMatrixs = left + right;
stk.push(newMatrixs);
MATRIXS[newMatrixs] = make_pair(MATRIXS[left].first,
MATRIXS[right].second);
}
}
return numMulti;
}
int main(int argc, char const *argv[])
{
#ifndef ONLINE_JUDGE
freopen("442_i.txt", "r", stdin);
freopen("uva_o.txt", "w", stdout);
#endif
int numMatrix;
cin >> numMatrix;
// 读入 first part
string matrixName;
int row, column;
while (numMatrix--) {
cin >> matrixName >> row >> column;
MATRIXS[matrixName] = make_pair(row, column);
}
// 读入 second part
string line;
cin.ignore(); // cin 之后接 getline 要有 ignore
while (!cin.eof()) {
getline(cin, line);
// 这里有可能会出现空行, 空行不做处理, 否则走到流程里, 会输出 0, 导致无限 WA
// 这种请况既然没有说明, 就不该出现在 test case 里
if (line.size() == 0) continue;
int numMulti = -1;
numMulti = getNumMulti(line);
if (numMulti == -1) {
cout << "error" << endl;
} else {
cout << numMulti << endl;
}
}
return 0;
}
环境: C++ 4.5.3 - GNU C++ Compiler with options: -lm -lcrypt -O2 -pipe -DONLINE_JUDGE
==============================================================
附: 考虑比较完善的代码, 即当括号不匹配时会返回 error; 括号内为空时, 也会返回 error; 奈何题目根本不要求这么多.
# include <iostream>
# include <string>
# include <cstdio>
# include <cstring>
# include <vector>
# include <algorithm>
# include <cctype>
# include <iterator>
# include <assert.h>
# include <map>
# include <stack>
using namespace std;
map< string, pair<int, int> > MATRIXS;
// 算法: 碰到 非) 就进栈, 碰到 ) 就看栈顶是不是匹配的 (, 是就出栈;
// 连续碰到两个 矩阵 名, 就相乘, 把结果入栈
// 若只碰到一个矩阵, 则入栈
// 计算两个矩阵相乘需要的相乘元素个数, 若两个矩阵不能相乘, 则返回 -1
// 两矩阵可以相乘是指 left 的 column 要等于 right 的 row
// 两矩阵的总乘数是指 left.row * left.column * right.columnl
int getNumMulti(const string left, const string right) {
if (MATRIXS.find(left) == MATRIXS.end() ||
MATRIXS.find(right) == MATRIXS.end()) {
return -1;
}
pair<int, int> matrixsLeft = MATRIXS[left];
pair<int, int> matrixsRight = MATRIXS[right];
if (matrixsLeft.second == matrixsRight.first) {
return matrixsLeft.first * matrixsLeft.second * matrixsRight.second;
} else {
return -1;
}
}
// 计算需要进行相乘的次数, 返回 -1 表示出错了
int getNumMulti(const string& line) {
// 只有一位, 用这个是一个有效矩阵, 则返回 0
if (line.size() == 1) {
string matrix(1, line[0]);
if (MATRIXS.find(matrix) != MATRIXS.end()) {
return 0;
} else {
return -1;
}
}
int numMulti = 0;
stack<string> stk;
for (int i=0; i<line.size(); i++) {
// 没有碰到 ), 则直接入栈; 若碰到了, 则取栈顶两个元素进行操作
if (line[i] != ')') {
string matrix(1, line[i]);
stk.push(matrix);
} else {
// 如果出现 () 或 A), 则返回 false
if (stk.size() < 2) return -1;
while (stk.top() != "(") {
string right = stk.top();
stk.pop();
string left = stk.top();
stk.pop();
if (left == "(") {
numMulti += 0;
stk.push(right);
break;
} else {
int num = getNumMulti(left, right);
if (num == -1) return -1;
numMulti += num;
string newMatrixs = left + right;
stk.push(newMatrixs);
MATRIXS[newMatrixs] = make_pair(MATRIXS[left].first,
MATRIXS[right].second);
}
}
}
}
if (!stk.empty()) {
// 要考虑 AB 这样不带括号的情况
while (stk.top() != "(") {
if (stk.size() == 1) {
stk.pop();
break;
}
string right = stk.top();
stk.pop();
string left = stk.top();
stk.pop();
if (left == "(") {
return -1;
} else {
int num = getNumMulti(left, right);
if (num == -1) return -1;
numMulti += num;
string newMatrixs = left + right;
stk.push(newMatrixs);
MATRIXS[newMatrixs] = make_pair(MATRIXS[left].first,
MATRIXS[right].second);
}
}
}
return numMulti;
}
int main(int argc, char const *argv[])
{
#ifndef ONLINE_JUDGE
freopen("442_i.txt", "r", stdin);
freopen("uva_o.txt", "w", stdout);
#endif
int numMatrix;
cin >> numMatrix;
// 读入 first part
string matrixName;
int row, column;
while (numMatrix--) {
cin >> matrixName >> row >> column;
MATRIXS[matrixName] = make_pair(row, column);
}
// 读入 second part
string line;
cin.ignore(); // cin 之后接 getline 要有 ignore
while (!cin.eof()) {
getline(cin, line);
// 这里有可能会出现空行, 空行不做处理, 否则走到流程里, 会输出 0, 题目太SB了,
// 这种请况既然没有说明, 就不该出现在 test case 里
if (line.size() == 0) continue;
int numMulti = -1;
numMulti = getNumMulti(line);
if (numMulti == -1) {
cout << "error" << endl;
} else {
cout << numMulti << endl;
}
}
return 0;
}