poj2246 Matrix Chain Multiplication （棧）

題意：給定A-Z若干矩陣。以及一些表達式，根據表達式求出所有矩陣相乘的總次數。若不符合要求輸出“error”

 

 

思路：正則表達式，需要理解矩陣相乘的一些性質。

 

學習點：

1、棧 STL

頭文件 #include <stack>

定義： stack<類型> s；

操作： empty、size、top、pop、push

 

本題在運用棧時，其中的類型竟然可以使一種結構體。開眼界了！詳細見代碼註釋。

 

源代碼：

#include <iostream> #include <string> #include <stack> using namespace std; struct matrix { int nRow, nCol; }; matrix nM[30]; string nExp; int matNum; bool nFlag; long ans; void funcInput(); void funcSolve(); int main() { funcInput(); funcSolve(); return 0; } void funcInput() { string nName; cin >> matNum; for (int i = 0; i != matNum; ++i) { cin >> nName; int nIndex = nName[0] - 'A'; cin >> nM[nIndex].nRow >> nM[nIndex].nCol; } } void funcSolve() { stack<char> e; //±í´ïʽջ stack<matrix> m; //Éæ¼°µÄ¾ØÕóÕ»£¬½á¹¹ÌåµÄʹÓã¬ÐÂ֪ʶµã£¡ while (cin >> nExp) { // e.clear(); m.clear(); nFlag = true; ans = 0; int len = nExp.size(); for (int i = 0; i != len; ++i) { if (nExp[i] == '(') e.push(nExp[i]); else if (nExp[i] >= 'A' && nExp[i] <= 'Z') m.push(nM[nExp[i]-'A']); else { if (e.empty()) { nFlag = false; break; } e.pop(); matrix a = m.top(); m.pop(); matrix b = m.top(); m.pop(); if (a.nRow != b.nCol) { nFlag = false; break; } ans += a.nRow * a.nCol * b.nRow; a.nRow = b.nRow; m.push(a); } } if (nFlag) cout << ans << endl; else cout << "error" << endl; } }