leetcode 394. 字符串解码 medium

leetcode 394. 字符串解码 medium          

题目描述：

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

编码规则为: 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".

解题思路：

字符串题都比较无聊...orz，就是考验个细心和api熟不熟感觉

方法一：直接用栈

方法二：递归（说实话，递归的解法我还真不会）

代码：

func decodeString(s string) string {
	stk := stack[byte]{}
	for i := 0; i < len(s); i++ {
		c := s[i]
		if c != ']' {
			stk.push(c)
			continue
		}
		word := ""
		for stk.top() != '[' {
			word = string(stk.pop()) + word
		}
		stk.pop()
		strNum := ""
		for !stk.empty() && IsNumber(stk.top()) {
			strNum = string(stk.pop()) + strNum
		}

		num, _ := strconv.ParseInt(strNum, 10, 32)
		one := strings.Repeat(word, int(num))
		for j := 0; j < len(one); j++ {
			stk.push(one[j])
		}
	}

	// res := ""
	// for !stk.empty() {
	// 	res = string(stk.pop()) + res
	// }
	return string(stk)
}

func IsNumber(b byte) bool {
	return b >= '0' && b <= '9'
}

type stack[T any] []T

func (s stack[T]) empty() bool {
	return len(s) == 0
}

func (s *stack[T]) push(val T) {
	*s = append(*s, val)
}

func (s *stack[T]) pop() T {
	old := *s
	n := len(old)
	res := old[n-1]
	*s = old[:n-1]
	return res
}

func (s stack[T]) top() T {
	return s[len(s)-1]
}

// 直接用栈
class Solution {
public:
    string decodeString(string s) {
        
        stack<char> stk;
        for(auto c:s){
            if(c !=']') stk.push(c);
            else{
                string one;
                while(stk.top() != '['){
                    one = stk.top()+one;
                    stk.pop();
                }
                stk.pop(); // pop掉 [
                string num;
                
                while(stk.size() && isdigit(stk.top())){
                    num=stk.top()+num;
                    stk.pop();
                }
               /* while(stk.size() && stk.top()>='0' && stk.top()<='9'){// isdigit()
                    num= stk.top()+num;
                    stk.pop();
                }
                
                */
                int k=stoi(num);
                string s1;
                for(int i=0;i<k;++i){
                    s1 += one;
                }
                
                for(char chr:s1) stk.push(chr);
            }
        }
        
        string res;
        while(!stk.empty()){
            res = stk.top()+res;
            stk.pop();
        }
        
        return res;
        
    }
};




//递归
class Solution {
public:
    string decodeString(string s) {
        int i = 0;
        return decode(s, i);
    }
    string decode(string s, int& i) {
        string res = "";
        int n = s.size();
        while (i < n && s[i] != ']') {
            if (s[i] < '0' || s[i] > '9') {
                res += s[i++];
            } else {
                int cnt = 0;
                while (s[i] >= '0' && s[i] <= '9') {
                    cnt = cnt * 10 + s[i++] - '0';
                }
                ++i;
                string t = decode(s, i);
                ++i;
                while (cnt-- > 0) {
                    res += t;
                }
            }
        }
        return res;
    }
};