PAT B1078 字符串压缩与解压 (20分)

题目链接：https://pintia.cn/problem-sets/994805260223102976/problems/994805262018265088

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

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

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

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

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

样例输入
C
TTTTThhiiiis isssss a tesssst CAaaa as

样例输出
5T2h4is i5s a3 te4st CA3a as

代码

#include <iostream>
#include <string>
using namespace std;

int main() {
	string str;
	char c;
	scanf("%c", &c);
	getchar();
	getline(cin, str);
	int len = str.size();
	int num;
	if(c == 'C') {
		num = 1;
		char pre = str[0];
		for(int i = 1; i <= len; i++) {
			if(str[i] != pre) {
				if(num > 1)
					printf("%d", num);
				printf("%c", str[i - 1]);
				num = 1;
				pre = str[i];
			}
			else
				num++;
		}
	}
	if(c == 'D') {
		num = 0;
		for(int i = 0; i < len; i++) {
			while(str[i] >= '0' && str[i] <= '9') {
				num = num * 10 + str[i] - '0';
				i++;
			}
			if(num)
				while(num--)
					printf("%c", str[i]);
			else
				printf("%c", str[i]);
			num = 0;
		}
	}
	return 0;
}