PTA 1078 字符串压缩与解压

题目如下：

文本压缩有很多种方法，这里我们只考虑最简单的一种：把由相同字符组成的一个连续的片段用这个字符和片段中含有这个字符的个数来表示。例如 ccccc 就用 5c 来表示。如果字符没有重复，就原样输出。例如 aba 压缩后仍然是 aba。
解压方法就是反过来，把形如 5c 这样的表示恢复为 ccccc。
本题需要你根据压缩或解压的要求，对给定字符串进行处理。这里我们简单地假设原始字符串是完全由英文字母和空格组成的非空字符串。

输入格式：

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

输出格式：

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

输入样例 1：

C
TTTTThhiiiis isssss a tesssst CAaaa as

输出样例 1：

5T2h4is i5s a3 te4st CA3a as

输入样例 2：

D
5T2h4is i5s a3 te4st CA3a as10Z

输出样例 2：

TTTTThhiiiis isssss a tesssst CAaaa asZZZZZZZZZZ

解题思路

这题的关键在于如何记录下字母出现的次数。这里我使用一个string用来专门保存输入的数字，将数字存入string中，直到下一次输入不为数字，将数组中的数字取出，利用函数atoi转换成int型作为count变量的值专门控制字符串的输出。每次遇到字符时都应当将count重新置为1以保证不出错。
字符串的压缩更加简单，利用循环判断并记录是否为叠字，如果是则count++，直到不是叠字的时候利用cout输出即可。

源代码

#include <iostream>
using namespace std;
int main() {
    char t;
    cin >> t;
    getchar();
    string str, num;
    getline(cin, str);
    int count = 1;
    if (t == 'D') 
	{
        for (int i = 0; i < str.length(); i++) 
		{
            if (str[i] >= '0' && str[i] <= '9') 
			{
                num += str[i];
            }
			else
			{
                if (num.length() > 0)
				{
					count = stoi(num);
				}
                while(count--) 
				{
					cout << str[i];
				}
                count = 1;
                num = "";
            }
        }
    } 
	else if (str.length() != 0)
	{
        char pre = str[0];
        for (int i = 1; i < str.length(); i++) 
		{
            if (str[i] == pre) 
			{
                count++;
            }
			else
			{
                if (count >= 2)
				{
					cout << count;
				} 
                cout << pre;
                count = 1;
                pre = str[i];
            }
        }
        if (count >= 2)
		{
			cout << count;
		}
        cout << pre;
    }
    return 0;
}

运行结果