计算原子的个数

import java.util.*;

// 栈 + Map计数
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String s = in.next();
        char[] c = s.toCharArray();
        Deque<Map<String, Integer>> stack = new LinkedList<>();
        int n = c.length;
        for (int i = 0; i < n;) {
            if (Character.isUpperCase(c[i])) {
                int j = i + 1;
                while (j < n && Character.isLowerCase(c[j])) j++;
                String key = s.substring(i, j);
                int num = 0;
                while (j < n && Character.isDigit(c[j])) {
                    num = num * 10 + c[j] - '0';
                    j++;
                }
                i = j;
                num = num == 0 ? 1 : num;
                Map<String, Integer> map = new HashMap<>();
                map.put(key, num);
                stack.push(map);
            } else if (c[i] == '(') {
                Map<String, Integer> map = new HashMap<>();
                map.put("(", -1);
                stack.push(map);
                i++;
            } else { // 遇到 ) 要合并 () 里面的并找到与之匹配的 (
                int num = 0, j = i + 1;
                while (j < n && Character.isDigit(c[j])) {
                    num = num * 10 + c[j] - '0';
                    j++;
                }
                i = j;
                num = num == 0 ? 1 : num;
                Map<String, Integer> map = new HashMap<>();
                while (!stack.peek().containsKey("(")) {
                    Map<String, Integer> mp = stack.pop();
                    for (String key : mp.keySet()) {
                        map.merge(key, mp.get(key) * num, Integer::sum);
                    }
                }
                stack.pop();
                stack.push(map);
            }
        }

        Map<String, Integer> map = new HashMap<>(); // 合并栈内剩余的
        while (!stack.isEmpty()) {
            Map<String, Integer> mp = stack.pop();
            for (String key : mp.keySet()) {
                map.merge(key, mp.get(key), Integer::sum);
            }
        }
        List<String> list = new ArrayList<>(map.keySet()); // 字典序
        Collections.sort(list);

        StringBuilder sb = new StringBuilder();
        for (String key : list) {
            sb.append(key);
            int val = map.get(key);
            if (val > 1) sb.append(val);
        }
        System.out.println(sb.toString());
    }
}