猿辅导笔试题：辅助栈 解压字符串

1、题目描述：

猿辅导APP需要下发一些宣传文本给学生，工程师们使用了一种字符压缩算法，为简单起见，假设被压缩的字符全部为大写字母序列，A,B,C,D…Z,压缩规则如下： 1.AAAB可以压缩为A3B (单字符压缩不加括号) 2.ABABA可以压缩为(AB)2A （多字符串压缩才加括号）
输入数据保证不会出现冗余括号，且表示重复的数字一定合法且大于1，即不会出现： 1.（A)2B ------- （应为：A2B） 2. ((AB))2C,-----(应为：（AB)2C ) 3. （A)B ----- （应为：AB） 4. A1B，（AB)1C，（应为 AB，ABC）
注意：数字可能出现多位数即A11B或者(AB)10C或者A02这种情况。 A11B = AAAAAAAAAAAB (AB)10C = ABABABABABABABABABABC A02 = AA
数据分布： 对于60%的数据，括号不会出现嵌套，即不会有 ((AB)2C)2这种结构。 对于80%的数据，括号最多只嵌套一层，即不会有 (((AB)2C)2D)99 这种结构。 对于100%的数据，括号可以嵌套任意层。

输入描述:
第一行是正整数C(C <= 100)，表示下面有C组数据。之后C行，每行为一组数据，每组数据为一个字符串。

每个字符串由A-Z,数字0-9和(,)组成表示一个压缩后的串，保证输入数据一定合法且字符串长度小于50。

输出描述:
输出C行，每行对应一个数据的输出结果，表示压缩前的字符串，保证每个字符串展开后的长度不超过10^6。

示例1

输入
5
A11B
(AA)2A
((A2B)2)2G
(YUANFUDAO)2JIAYOU
A2BC4D2

输出
AAAAAAAAAAAB
AAAAA
AABAABAABAABG
YUANFUDAOYUANFUDAOJIAYOU
AABCCCCDD

2、题解：

建议同时做辅助栈 | 深度优先搜索DFS：力扣394. 字符串解码
辅助栈法：

构造一个辅助栈
对于待处理的字符串，从左到右扫描，
    1、遇到'('，就让stack添加一个空串''，用来保存(...)里的字符串，更新i，寻找下一个
    2、遇到')',就找其中的倍数num，然后top = stack.pop()表示取出保存的')'对应的'('之间的的字符串，然后更新栈顶元素，
    也即让栈顶元素+top * num
    3、遇到数字字符，就找全倍数num，更新栈顶字符串的值，也就是让栈顶的最后一个字符与(num - 1)相乘后再加上原栈顶的值
    4、遇到字母，就添加到栈顶就行，然后更新i寻找下一个
返回 stack[0],解释：到最后栈中只剩下一个字符串

Python代码如下：

def decodeString(s):
    #辅助栈
    res,multi,stack = '',0,['']
    i = 0
    while i < len(s):
        if s[i] == '(':
            stack.append('')
            # res = ''
            i += 1
        elif s[i] == ')':
            k = i + 1
            i += 1
            while i < len(s) and '0' <= s[i] <= '9':
                i += 1
            num = int(s[k:i])
            top = stack.pop()
            stack[-1] += top * num
        elif  '0' <= s[i] <= '9':
            k = i
            i += 1
            while i < len(s) and '0' <= s[i] <= '9':
                i += 1
            num = int(s[k:i])
            stack[-1] += stack[-1][-1] * (num - 1)
        else:
            stack[-1] += s[i]
            i += 1
    return stack[0]

n = int(input())
for _ in range(n):
    s = input()
    print(decodeString(s))

3、复杂度分析：

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