LeetCode 726. 原子的数量 （困难）

726. 原子的数量 （困难）

给定一个化学式formula（作为字符串），返回每种原子的数量。 原子总是以一个大写字母开始，接着跟随0个或任意个小写字母，表示原子的名字。
如果数量大于 1，原子后会跟着数字表示原子的数量。如果数量等于 1 则不会跟数字。例如，H2O 和 H2O2 是可行的，但 H1O2
这个表达是不可行的。 两个化学式连在一起是新的化学式。例如 H2O2He3Mg4 也是化学式。
一个括号中的化学式和数字（可选择性添加）也是化学式。例如 (H2O2) 和 (H2O2)3 是化学式。
给定一个化学式，输出所有原子的数量。格式为：第一个（按字典序）原子的名子，跟着它的数量（如果数量大于
1），然后是第二个原子的名字（按字典序），跟着它的数量（如果数量大于 1），以此类推。

实例1：

输入: formula = “H2O” 输出: “H2O” 解释: 原子的数量是 {‘H’: 2, ‘O’: 1}。

分析： 这种类型的题目 就是考察代码逻辑能力和严谨能力，比较容易出错，代码量也比较大

化学式分成2种，一种普通化学式没有括号，一种带有括号的化学式

题解：

我们可以采用栈的思想去写，因为有括号，如果遇到右括号了，我们就要去匹配。

思路：

1、如何存储： 采用栈，栈里面采用map来存储每一个括号里面的字母个数
2、如果遇到左括号，我们往栈顶添加一个空map，指针i往后走一步
3、如果遇到右括号，说明已经遇到一对完整的括号了，那么括号了里面的字母需要统计，这时候栈顶的map存储的就是当前括号里的字母，然后我们需要把右括号后面的数字拿到，Ca(OH)2 就是拿这个2，然后我们需要把当前栈顶的map合并到前一个map里，合并的时候记得乘以刚刚拿的数字。

代码如下：

else if(formula.substr(i,1)==")")
{
    map<string,int>k=sk.top();
    sk.pop();
    int j=++i;
    int num=0;
    while(j<len&&formula[j]>='0'&&formula[j]<='9')
    {
        num=num*10+formula[j]-'0';
        j++;
    }
    for(auto &x:k)
    {
        if(num==0)num=1;
        sk.top()[x.first]+=num*(x.second);
    }
    i=j;
}

如果遇到大写字母，我们就往后取，取到不是小写字母位置，然后拼接元素
加入到栈顶的map里即可

按照上述的顺序编码能解决此题、

AC代码：

class Solution {
public:
    string countOfAtoms(string formula) {
        stack<map<string,int>>sk;
        map<string,int>ma;
        sk.push(ma);
        int len=formula.length();
        for(int i=0;i<len;)
        {
            if(formula.substr(i,1)=="(")
            {
                sk.push(ma);
                i++;
            }
            else if(formula.substr(i,1)==")")
            {
                map<string,int>k=sk.top();
                sk.pop();
                int j=++i;
                int num=0;
                while(j<len&&formula[j]>='0'&&formula[j]<='9')
                {
                    num=num*10+formula[j]-'0';
                    j++;
                }
                for(auto &x:k)
                {
                    if(num==0)num=1;
                    sk.top()[x.first]+=num*(x.second);
                }
                i=j;
            }
            else
            {//如果是字母 
                string zhi=formula.substr(i,1);
                int j=++i;
                while(j<len&&formula[j]>='a'&&formula[j]<='z')
                {
                    j++;
                } 
                zhi+=formula.substr(i,j-i);
                //cout<<i<<" "<<j<<" "<<j-i<<endl;
                //cout<<formula.substr(i,j-i)<<endl;
                int num=0;
                while(j<len&&formula[j]>='0'&&formula[j]<='9')
                {
                    num=num*10+formula[j]-'0';
                    j++;
                }
                sk.top()[zhi]+=(num==0?1:num);
                i=j;
            }
        }
        string ans="";
        map<string,int>ans2;
        while(!sk.empty())
        {
            ma=sk.top();
            sk.pop();
            for(auto &x:ma)
            {
                ans2[x.first]=x.second;
    //			cout<<x.first<<" "<<x.second<<endl;
            }
    //		cout<<"---------------------"<<endl;
        }
        for(auto &x:ans2)
        {
            ans+=x.first;
            if(x.second>1)
            {
                ans+=to_string(x.second);
            }
        }
        
        return ans;
    }
};