记录 - 394. 字符串解码

191 阅读1分钟

        给定一个经过编码的字符串,返回它解码后的字符串。 编码规则为: k[encoded_string],表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。 你可以认为输入字符串总是有效的;输入字符串中没有额外的空格,且输入的方括号总是符合格式要求的。 此外,你可以认为原始数据不包含数字,所有的数字只表示重复的次数 k ,例如不会出现像 3a 或 2[4] 的输入。 

示例: 

s = "3[a]2[bc]", 返回 "aaabcbc". 

s = "3[a2[c]]", 返回 "accaccacc". 

s = "2[abc]3[cd]ef", 返回 "abcabccdcdcdef".  


     public static String decodeString(String s) {
        if(s.equals("")) return s;
        Stack<Integer> stackMulti = new Stack<>();
        Stack<String> source = new Stack<>();
        StringBuilder result = new StringBuilder();
        int multiValue = 0;
        for(int i =0; i < s.length(); i++){
            char unit = s.charAt(i);
            if(unit >= '0' && unit <= '9'){
                multiValue = multiValue*10 +  Integer.valueOf(String.valueOf(unit));
            } else if(unit == '['){
                if (multiValue != 0){
                    stackMulti.push(multiValue);
                    multiValue = 0;
                }
                source.push("");
            } else if(unit == ']'){
                StringBuilder stringBuilder = new StringBuilder();
                    int multi = stackMulti.pop();
                    String stringUnitInner = source.pop();
                    for (int j = 0; j < multi; j++) {
                        stringBuilder.append(stringUnitInner);
                    }
                    if (source.size() > 0) {
                        source.push(source.pop() + stringBuilder.toString());
                    } else {
                        result.append(stringBuilder);
                    }
            } else {
           if (stackMulti.size() > 0)
                    source.push(source.pop()+ unit);
                else
                    result.append(unit);
            }
        }
        return result.toString();
    }