394. Decode String

Description

Given an encoded string, return it's decoded string.

The encoding rule is: k[encoded_string], where the encoded_string inside the square brackets is being repeated exactly k times. Note that k is guaranteed to be a positive integer.

You may assume that the input string is always valid; No extra white spaces, square brackets are well-formed, etc.

Furthermore, you may assume that the original data does not contain any digits and that digits are only for those repeat numbers, k. For example, there won't be input like 3a or 2[4].

Examples:

s = "3[a]2[bc]", return "aaabcbc".
s = "3[a2[c]]", return "accaccacc".
s = "2[abc]3[cd]ef", return "abcabccdcdcdef".

Solution

DFS

这道题让我们把一个按一定规则编码后的字符串解码成其原来的模样,编码的方法很简单,就是把重复的字符串放在一个中括号里,把重复的次数放在中括号的前面,注意中括号里面有可能会嵌套中括号,这题可以用递归和迭代两种方法来解。

我们首先来看递归的解法,我们把一个中括号中的所有内容看做一个整体,一次递归函数返回一对中括号中解码后的字符串。给定的编码字符串实际上只有四种字符,数字,字母,左中括号,和右中括号。那么我们开始用一个变量i从0开始遍历到字符串的末尾,由于左中括号都是跟在数字后面,所以我们首先遇到的字符只能是数字或者字母,如果是字母,我们直接存入结果中,如果是数字,我们循环读入所有的数字,并正确转换,那么下一位非数字的字符一定是左中括号,我们指针右移跳过左中括号,对之后的内容调用递归函数求解,注意我们循环的停止条件是遍历到末尾和遇到右中括号,由于递归调用的函数返回了子中括号里解码后的字符串,而我们之前把次数也已经求出来了,那么循环添加到结果中即可,参见代码如下:

注意递归需要返回当前层处理到哪个index之前,否则上层无法得知改从哪里开始继续处理。换言之,上层需要知道下层处理到什么地方了,即’]'的位置。我这里返回的idx代表该从哪里开始,即']'的后一个位置。

class Solution {
    public String decodeString(String s) {
        return decode(s.toCharArray(), 0).str;
    }
    
    public Pair decode(char[] arr, int start) {
        StringBuilder sb = new StringBuilder();
        int i = start;
        
        while (i < arr.length) {
            if (Character.isDigit(arr[i])) {
                int count = 0;
                while (Character.isDigit(arr[i])) {
                    count = 10 * count + (arr[i++] - '0');
                }
                // arr[i] should be '[' now
                Pair sub = decode(arr, i + 1);
                while (count-- > 0) {
                    sb.append(sub.str);
                }
                i = sub.idx;
            } else if (arr[i] == ']') {     // consume one ']'
                ++i;
                break;
            } else {
                sb.append(arr[i++]);
            }
        }
        
        
        return new Pair(sb.toString(), i);
    }
    
    class Pair {
        String str;
        int idx;
        
        public Pair(String str, int idx) {
            this.str = str;
            this.idx = idx;
        }
    }
}

Stack, time O(n), space O(n)

用Stack做更赏心悦目一点。

class Solution {
    public String decodeString(String s) {
        if (s == null || s.isEmpty()) {
            return s;
        }
        
        Stack<Integer> countStack = new Stack<>();  // peek() is curr level count
        Stack<String> strStack = new Stack<>();     // peek() is prev level str
        int i = 0;
        String res = "";    // store curr level str
        
        while (i < s.length()) {
            char c = s.charAt(i);
            
            if (Character.isDigit(c)) {
                int count = 0;
                // count might be over 1-digit length
                while (i < s.length() && Character.isDigit(s.charAt(i))) {
                    count = 10 * count + s.charAt(i) - '0';
                    ++i;
                }
                
                countStack.push(count);
            } else if (c == '[') {  // enter a new level, push and clear res
                strStack.push(res);
                res = "";
                ++i;
            } else if (c == ']') {  // curr level ends, merge it into prev level
                StringBuilder sb = new StringBuilder();
                sb.append(strStack.pop());
                int count = countStack.pop();
                
                while (count-- > 0) {
                    sb.append(res);
                }
                
                res = sb.toString();
                ++i;
            } else {
                res += c;
                ++i;
            }
        }
        
        return res;
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,311评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,339评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,671评论 0 342
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,252评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,253评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,031评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,340评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,973评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,466评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,937评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,039评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,701评论 4 323
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,254评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,259评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,485评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,497评论 2 354
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,786评论 2 345

推荐阅读更多精彩内容