LeetCode 726. 原子的数量

给定一个化学式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;
    }
};

 

转载于:https://www.cnblogs.com/hlk09/p/9735103.html