题目描述
给定一个化学式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}。
注意:
所有原子的第一个字母为大写,剩余字母都是小写。
formula的长度在[1, 1000]之间。
formula只包含字母、数字和圆括号,并且题目中给定的是合法的化学式。
代码
class Solution {
public String countOfAtoms(String formula) {
int N = formula.length();
Stack<Map<String, Integer>> stack = new Stack();
stack.push(new TreeMap());
for (int i = 0; i < N;) {
if (formula.charAt(i) == '(') {
//开始一颗新的树放进新的栈桢 push进去新元素
stack.push(new TreeMap());
i++;
} else if (formula.charAt(i) == ')') {
Map<String, Integer> top = stack.pop();
//获得括号后的出现次数
int iStart = ++i, multiplicity = 1;
while (i < N && Character.isDigit(formula.charAt(i))) i++;
if (i > iStart) multiplicity = Integer.parseInt(formula.substring(iStart, i));
//对栈顶的这棵树里面的每一个元素 都对出现次数进行乘法
for (String c: top.keySet()) {
int v = top.get(c);
stack.peek().put(c, stack.peek().getOrDefault(c, 0) + v * multiplicity);
}
} else {
//获得元素名称
int iStart = i++;
while (i < N && Character.isLowerCase(formula.charAt(i))) i++;
String name = formula.substring(iStart, i);
//获得元素名称后紧跟着的数字
iStart = i;
while (i < N && Character.isDigit(formula.charAt(i))) i++;
int multiplicity = i > iStart ? Integer.parseInt(formula.substring(iStart, i)) : 1;
//在栈顶的树结构 放进去(元素名称,出现次数)
stack.peek().put(name, stack.peek().getOrDefault(name, 0) + multiplicity);
}
}
StringBuilder ans = new StringBuilder();
for (String name: stack.peek().keySet()) {
ans.append(name);
int multiplicity = stack.peek().get(name);
if (multiplicity > 1) ans.append("" + multiplicity);
}
return new String(ans);
}
}