【Leetcode】394. Decode String

题目地址：

https://leetcode.com/problems/decode-string/

给定一个字符串，要求按照如下方式解码：
1、每次遇到形如$k[s]$的形式，就等价替换为$k$个$s$的重复；
2、按照1的方式等价替换后，对替换完后的字符串再接着进行同样的1的操作，直到替换不下去为止，最后返回最终结果。

题目有些像括号匹配的问题，于是想到用栈来做。这里可以采用这样的方式：用一个栈来存放重复次数，另设一个StringBuilder直接当成栈来用，每次遇到了形如$k[s]$的形式，就直接解析出来，append到StringBuilder后面。代码如下：

import java.util.Deque;
import java.util.LinkedList;

public class Solution {
    public String decodeString(String s) {
    	// 判空
        if (s == null || s.length() == 0) {
            return s;
        }
    	// 设一个栈，来存放每一层的字符串需要重复的次数
        Deque<Integer> numStack = new LinkedList<>();
        StringBuilder res = new StringBuilder();
        for (int i = 0; i < s.length(); i++) {
        	// 如果遇到了数字，就直接解析出来这个数字是多少，然后加入栈中
            if (Character.isDigit(s.charAt(i))) {
                int n = 0;
                while (i < s.length() && Character.isDigit(s.charAt(i))) {
                    n *= 10;
                    n += s.charAt(i) - '0';
                    i++;
                }
                numStack.push(n);
            }
            // 解析完数字后，将字符直接加到res后面
            res.append(s.charAt(i));
            // 如果遇到了右括号，说明需要把该重复的字符串解析出来了
            if (s.charAt(i) == ']') {
            	// 先寻找到最右边的左括号
                int lastLeft = res.length() - 1;
                while (res.charAt(lastLeft) != '[') {
                    lastLeft--;
                }
                // 分割出需要重复的那个字符串
                String lastWord = res.substring(lastLeft + 1, res.length() - 1);
                // 把最后一个形如"[*]"的字符串删掉
                res.setLength(lastLeft);
                // 把需要重复的字符串重复若干次append到后面
                res.append(lastWord.repeat(numStack.pop()));
            }
        }
        
        return res.toString();
    }
}

时空复杂度$O(n)$，$n$是最后解码出来的字符串长度。