【PAT乙级】字符串压缩与解压

题目描述：

文本压缩有很多种方法，这里我们只考虑最简单的一种：把由相同字符组成的一个连续的片段用这个字符和片段中含有这个字符的个数来表示。例如 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

 AC代码：

#include 
using namespace std;

void compression(string s);   //压缩字符串
void decompression(string s);   //解压字符串

int main()
{
    char ch;
    cin >> ch;   //输入一个字符
    getchar();   //吃回车
    string s;
    getline(cin,s);    //获取一行字符串
    switch(ch)  //判断是压缩还是解压缩
    {
        case 'C': compression(s); break;   //压缩字符串
        case 'D': decompression(s); break;  //解压字符串
        default: break;
    }
    return 0;
}

void compression(string s)   //压缩字符串
{
    for(int i = 0; i < s.length(); i++)   //遍历字符串
    {
        int cnt = 0;   //用来记录重复字符数量
        while(s[i] == s[i+1])   //判断是不是字符串中的重复字符
        {
            i++;
            cnt++;   
        }
        if(cnt != 0)
        {
            cout << cnt+1;   //压缩后的形式,先输出重复字符的个数
        }
        cout << s[i];   //再输出重复字符
    }
}

void decompression(string s)   //解压字符串
{
    for(int i = 0; i < s.length(); i++)   //遍历字符串
    {
        int cnt = 0;
        while(s[i] >= '0' && s[i] <= '9')
        {
            cnt = cnt*10 + s[i]-'0';  //统计被压缩字符串的个数,可能是俩位数
            i++;
        }
        for(int j = 0; j < cnt; j++)
        {
            cout << s[i];   //输出被压缩的字符
        } 
        if(cnt == 0)   //如果不是被压缩的字符则直接输出
        {
            cout << s[i];
        }
    }
}