算法初探LeetCode-字符串解码

开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第1天，点击查看活动详情

LeetCode394：字符串解码

给定一个经过编码的字符串，返回它解码后的字符串。

编码规则为: k[encoded_string]，表示其中方括号内部的 encoded_string 正好重复 k 次。注意 k 保证为正整数。

你可以认为输入字符串总是有效的；输入字符串中没有额外的空格，且输入的方括号总是符合格式要求的。

此外，你可以认为原始数据不包含数字，所有的数字只表示重复的次数 k ，例如不会出现像 3a 或 2[4] 的输入。

示例 1：

输入： s = "3[a]2[bc]"
输出： "aaabcbc"

示例 2：

输入： s = "3[a2[c]]"
输出： "accaccacc"

示例 3：

输入： s = "2[abc]3[cd]ef"
输出： "abcabccdcdcdef"

示例 4：

输入： s = "abc3[cd]xyz"
输出： "abccdcdcdxyz"

提示：

• 1 <= s.length <= 30
• s 由小写英文字母、数字和方括号 '[]' 组成
• s 保证是一个 有效 的输入。
• s 中所有整数的取值范围为 [1, 300]

思路分析

分析题意，本题当前采用递归方式，关键点在于将 [ 和 ] 分别作为递归的开启与终止条件：

• 当 s[i] == ']' 时，返回当前括号内记录的 res 字符串与 ] 的索引 i （更新上层递归指针位置）；
• 当 s[i] == '[' 时，开启新一层递归，记录此 [...] 内字符串 tmp 和递归后的最新索引 i，并执行 res + multi * tmp 拼接字符串;
• 遍历完毕后返回 res。

算法代码

public String decodeString(String s) {
    return dfs(s, 0)[0];
}
// 递归处理
private String[] dfs(String s, int i) {
    StringBuilder res = new StringBuilder();
    int multi = 0;
    while(i < s.length()) {
        if(s.charAt(i) >= '0' && s.charAt(i) <= '9') 
            multi = multi * 10 + Integer.parseInt(String.valueOf(s.charAt(i))); 
        else if(s.charAt(i) == '[') {
            String[] tmp = dfs(s, i + 1);
            i = Integer.parseInt(tmp[0]);
            while(multi > 0) {
                res.append(tmp[1]);
                multi--;
            }
        }
        else if(s.charAt(i) == ']') 
            return new String[] { String.valueOf(i), res.toString() };
        else 
            res.append(String.valueOf(s.charAt(i)));
        i++;
    }
    return new String[] { res.toString() };
} 

结果

算法复杂度

• 时间复杂度：$O(N)$
• 空间复杂度：$O(N)$

在掘金(JUEJIN)一起进步，一起成长！