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. 当遇到’('时,对括号内的元素进行递归
2. 当遇到’)'时，携带map出栈（map内原子个数还需要乘括号后出现的数字，如果没有，默认为1），回到上一级函数，并且将返回的map进行合并
3. 在递归函数中，遇到字母，即需要处理原子，统计个数，使用treemap记录该原子出现次数

代码

class Solution {
    int next=-1;
    public String countOfAtoms(String formula) {

        StringBuilder builder = new StringBuilder();
        Map<String, Integer> map = count(formula, 0);
        for (Map.Entry<String, Integer> entry : map.entrySet()) {
            builder.append(entry.getKey()).append(entry.getValue()==1?"":entry.getValue());

        }
        return builder.toString();

    }
    public Map<String,Integer> merge(Map<String,Integer> a,Map<String,Integer> b) {
        for (Map.Entry<String,Integer> c:b.entrySet())
        {
            a.put(c.getKey(),a.getOrDefault(c.getKey(),0)+c.getValue());
        }
        return a;
    }
    public Map<String,Integer> count(final String formula, int l) {
        Map<String, Integer> map = new TreeMap<String, Integer>();
        for (int i=l;i<formula.length();){
            if(formula.charAt(i)=='(')
            {
                Map<String, Integer> count = count(formula, i + 1);
                merge(map,count);
                i=next;
            }else if(formula.charAt(i)==')'){//后面没有跟数字，默认值是1
                if(i+1>=formula.length()||!Character.isDigit(formula.charAt(i+1)))//没有跟数字
                    next=i+1;
                else {//把数字提取出来
                    i++;
                    int s=i;
                    while (i<formula.length()&&Character.isDigit(formula.charAt(i)))
                        i++;
                    int cur=Integer.parseInt(formula.substring(s,i));
                    map.replaceAll((k,v)->v*cur);
                    next=i;
                }
                return map;

            } else if(Character.isUpperCase(formula.charAt(i))) {
//遇到大写字母，如果后面有小写的也归于此原子
                int s = i;
                i++;
                while (i < formula.length() && Character.isLowerCase(formula.charAt(i)))
                    i++;
                String key = formula.substring(s, i);
                int cur;
                if (i >= formula.length() || !Character.isDigit(formula.charAt(i)))
                    cur = 1;
                else {
                    s = i;
                    while (i < formula.length() && Character.isDigit(formula.charAt(i)))
                        i++;
                    cur = Integer.parseInt(formula.substring(s, i));
                }
                map.put(key, map.getOrDefault(key, 0) + cur);
            }
        }
    return map;
    }
}