LeetCode 726. 原子的数量(栈)

题目描述
给定一个化学式formula（作为字符串），返回每种原子的数量。

原子总是以一个大写字母开始，接着跟随0个或任意个小写字母，表示原子的名字。

如果数量大于 1，原子后会跟着数字表示原子的数量。如果数量等于 1 则不会跟数字。例如，H2O 和 H2O2 是可行的，但 H1O2 这个表达是不可行的。两个化学式连在一起是新的化学式。例如 H2O2He3Mg4 也是化学式。一个括号中的化学式和数字（可选择性添加）也是化学式。例如 (H2O2) 和 (H2O2)3 是化学式。

给定一个化学式 formula ，返回所有原子的数量。格式为：第一个（按字典序）原子的名字，跟着它的数量（如果数量大于 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}。
示例 4：
输入：formula = “Be32”
输出：“Be32”

提示：
1 <= formula.length <= 1000
formula 由小写英文字母、数字 ‘(’ 和 ‘)’ 组成。
formula 是有效的化学式。

参考链接

从后往前遍历比从前往后遍历好理解些。在化学表达式中有括号嵌套的情况，且最里面的括号次数要乘以该括号外层括号的个数，比如样例"K4(ON(SO3)2)2"，S的个数就是2*2=4得到的。因此在脱掉括号的时候如果是“）” 那么cnt要乘以“）”外的数字，需要累乘，如果遇到“（” 需要把“）”括号旁的数字除掉。
如果遇到的是大写字母，那么需要对字符串翻转，注意字符串有可能一个字母，有可能两个，取决于化学元素组成。

class Solution {
public:
    map<string,int>zz;
    string countOfAtoms(string formula) {
        stack<int>ha;
        string s="",d="";
        int cnt=1;
        for(auto i=formula.rbegin();i!=formula.rend();++i)
        {  //从后往前遍历
            if(isdigit(*i))  d=*i+d;
            else if(*i==')')
            {
                int v=d==""?1:stoi(d);
                d="";
                cnt*=v;          //更新倍数
                ha.push(v);      //入栈
            }
            else if(*i=='(')
            {
                cnt/=ha.top();   //更新倍数
                ha.pop();        //出栈
            }
            else
            {
                s+=*i;
                if(*i>='A'&&*i<='Z')
                {
                    reverse(s.begin(),s.end());
                    zz[s]+=(d==""?1:stoi(d))*cnt;        
                    s="",d="";   //完成一次记录后清空记录的数值      
                }   
            }
        }
        for(auto i:zz)
        {
            s+=i.first;
            if(i.second>1) s+=to_string(i.second);
        }
        return s;
    }
};