Python基础练习 1078 字符串压缩与解压 (20 分)

本文已参与「新人创作礼」活动，一起开启掘金创作之路。

问题描述

　　文本压缩有很多种方法，这里我们只考虑最简单的一种：把由相同字符组成的一个连续的片段用这个字符和片段中含有这个字符的个数来表示。例如 ccccc 就用 5c 来表示。如果字符没有重复，就原样输出。例如 aba 压缩后仍然是 aba。

解压方法就是反过来，把形如 5c 这样的表示恢复为 ccccc。

本题需要你根据压缩或解压的要求，对给定字符串进行处理。这里我们简单地假设原始字符串是完全由英文字母和空格组成的非空字符串。

输入输出

• 输入格式

　　输入第一行给出一个字符，如果是 C 就表示下面的字符串需要被压缩；如果是 D 就表示下面的字符串需要被解压。第二行给出需要被压缩或解压的不超过 1000 个字符的字符串，以回车结尾。题目保证字符重复个数在整型范围内，且输出文件不超过 1MB。

• 输出格式

　　根据要求压缩或解压字符串，并在一行中输出结果。

• 样例输入

输入样例1：

C
TTTTThhiiiis isssss a tesssst CAaaa as

输入样例2：

D
5T2h4is i5s a3 te4st CA3a as10Z

• 样例输出

输出样例1：

5T2h4is i5s a3 te4st CA3a as

输出样例2：

TTTTThhiiiis isssss a tesssst CAaaa asZZZZZZZZZZ

Python版本 代码如下

这个题目要相对简单一些。可以首先做一个判断：如果是C，则压缩对应的字符串；如果是 D 就解压对应的字符串。其中ans用来存储运行的答案，con用来计数。

x = input()
s = input()
n = len(s)

ans=""
if x=="C":
    i=0
    while i<n:
        con=1
        j = i+1
        while j<n and s[i]==s[j]:
            con+=1
            j+=1
        if con==1:
            ans+=s[i]
        else:
            ans+=str(con)
            ans+=s[i]
        i = j
else:
    i=0
    while i<n:
        j = i+1
        if '0'<=s[i]<='9':
            while '0'<=s[j]<='9':
                j+=1
            ans += int(s[i:j])*s[j]
            i = j+1
        else:
            ans+=s[i]
            i=j
    
print(ans)