leetcode 394. 字符串解码 golang实现

描述

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

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

思路

利用栈实现

实现

type DecodeStringStruct struct{
	Times int   //
	Res string
}

func decodeString(s string) string {
	stack := make([]DecodeStringStruct, 0)
	var times int
	var res string

	m := map[byte]int{
		'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9,
	}

	for i := 0; i < len(s); i++ {
		b := s[i]
		if b >= '0' && b <= '9' {
			times = times*10 + m[b]
		} else if b == '[' {
			st := DecodeStringStruct{Times: times, Res: res}
			stack = append(stack, st)
			res = ""
			times = 0
		} else if b == ']' {
			st := stack[len(stack)-1]
			stack = stack[0 : len(stack)-1]
			res = st.Res + strings.Repeat(res, st.Times)
		} else {
			res = res + string(b)
		}
	}

	return res
}