给定一个化学式formula
(作为字符串),返回每种原子的数量。
原子总是以一个大写字母开始,接着跟随0个或任意个小写字母,表示原子的名字。
如果数量大于 1,原子后会跟着数字表示原子的数量。如果数量等于 1 则不会跟数字。例如,H2O 和 H2O2 是可行的,但 H1O2 这个表达是不可行的。
两个化学式连在一起是新的化学式。例如 H2O2He3Mg4 也是化学式。
一个括号中的化学式和数字(可选择性添加)也是化学式。例如 (H2O2) 和 (H2O2)3 是化学式。
给定一个化学式,输出所有原子的数量。格式为:第一个(按字典序)原子的名子,跟着它的数量(如果数量大于 1),然后是第二个原子的名字(按字典序),跟着它的数量(如果数量大于 1),以此类推。
示例 1:
输入: formula = "H2O" 输出: "H2O" 解释: 原子的数量是 {'H': 2, 'O': 1}。
示例 2:
输入: formula = "Mg(OH)2" 输出: "H2MgO2" 解释: 原子的数量是 {'H': 2, 'Mg': 1, 'O': 2}。
示例 3:
输入: formula = "K4(ON(SO3)2)2" 输出: "K4N2O14S4" 解释: 原子的数量是 {'K': 4, 'N': 2, 'O': 14, 'S': 4}。
类似于之前的“字符串解码”。需要注意的是类似A2BA这样子的化学式。我写的代码如下:
class Solution { public: string countOfAtoms(string formula) { stack<map<string, int>> s; map<string, int> curSet; for (int i = 0; i < formula.size(); ) { if (isupper(formula[i])) { int j = i + 1; string tmp, tmp1; tmp += formula[i]; if (j < formula.size() && islower(formula[j])) { tmp += formula[j++]; } if (j < formula.size() && isdigit(formula[j])) { while (j < formula.size() && isdigit(formula[j])) { tmp1 += formula[j]; j++; } } else { tmp1 = "1"; } curSet[tmp] += stoi(tmp1); i = j; } else if (formula[i] == '(') { s.push(curSet); curSet.clear(); i++; } else if (formula[i] == ')') { int j = i + 1; string tmp; while (j < formula.size() && isdigit(formula[j])) { tmp += formula[j]; j++; } int cnt = stoi(tmp); map<string, int>& top = s.top(); for (auto i = curSet.begin(); i != curSet.end(); i++) { top[i->first] += (i->second) * cnt; } curSet = top; s.pop(); i = j; } } string res; for (auto i = curSet.begin(); i != curSet.end(); i++) { res += i->first + (i->second != 1? (to_string(i->second)): ""); } return res; } };