字符串压缩算法（腾讯笔试题）

“ 将长度为n的字符串A通过设定的算法，转换为长度为m且小于n的字符串B，字符串B通过相应的算法，可以再次转换为字符串A，该算法称为字符串压缩算法。”

问题

 

小Q想要给他的朋友发送一个神秘字符串，但他发现字符串的长度过长，于是小Q发明了一种压缩算法对字符串重复的部分进行了压缩。

对于字符串内连续的m个相同的子串s将会压缩为m|s。例如字符串ABCABCABC会被压缩为[3|ABC]，现在小Q的同学收到了小Q发送过来的字符串，你能帮助还原字符串吗？

压缩算法描述：

被压缩的子串长度不超过1000，子串仅包含大写字母和[、]、|字符。算法扫描整个字符串，遇到连续相同的子串，对子串进行压缩，输出压缩后的格式字符串，若不是连续的子串，该子串原样输出。

压缩格式为：[count|子串]

其中，count为子串重复的次数，子串为连续重复的字符串。

示例1

输入压缩字符串
HG[3|B[2|CA]]F
输出解压后字符串
HGBCACABCACABCACAF

解读

本题难点在于括号内嵌套括号，需要从内向外生成与拼接字符串，可以使用递归结构对字符串进行解压缩。

编程思路：

定义x，y，z三个变量，默认值为-1，程序扫描字符串s时，分别记录字符'[、'|'、']'在字符串s的索引位置，在扫描过程中，若遇到字符']'，则停止扫描，提取处于字符'|'和字符']'范围内的重复子串，对提取的重复子串解压处理，并生成需要继续递归处理的字符串。若扫描结束x，y，z都为默认值，则直接返回s。

# 定义解码函数
def decode(s):
    i = 0
    x, y, z = -1, -1, -1
    # 遍历字符串s
    while i < len(s):
        # 记录'['的索引位置
        if s[i] == '[':
            x = i
        # 记录'|'的索引位置
        elif s[i] == '|':
            y = i
        # 记录']'的索引位置
        elif s[i] == ']':
            z = i
            # 扫描到']'字符时，跳出循环
            break
        i += 1
    # 处理重复的字符串
    if x != -1 and y != -1 and z != -1:
        # 从字符串s获取重复次数
        times = int(s[x + 1:y])
        # 从字符串s获取重复子串
        sub = s[y + 1:z]
        # 计算需要再次递归处理的字符串
        decode_str = s[:x] + times * sub + s[z + 1:]
        # 递归处理字符串
        return decode(decode_str)
    # 若没有重复的字符串，返回s
    return s  

if __name__=='__main__':
    
    print(decode("HG[3|B[2|CA]]F"), end='')

 

编程训练营APP

创新在线学习模式，学习编程不再半途而废

安卓手机应用商店

搜索编程训练营下载