PAT 乙级 1078 字符串压缩与解压 (20 分)

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

分析：

可以把压缩和解压的过程分为两个函数；Compression负责压缩，Decompress负责解压。压缩过程需要记录是否和前一个字符相同，可以使用一个计数器记录，当遇到不相同的字符时即输出个数和当前字符。解压过程需要从字符串中获得数字然后按个数输出当前的字符即可。每次循环计数器都会被重置为 1。

代码：

#include <iostream>
#include <string>

using namespace std;

char ch;
string str, ans = "";
int cnt = 1;

void Compression() {
    char pre = str[0];
    for (int i = 1; i < str.length(); i++) {
        if (str[i] == pre) {    // 如果和上一个字符相同
            cnt++;
        } else {
            if (cnt >= 2)
                cout << cnt;
            cout << pre;
            cnt = 1;
            pre = str[i];
        }
    }

    if (cnt >= 2)
        cout << cnt;
    cout << pre;
}

void Decompress() {
    for (int i = 0; i < str.length(); i++) {
        if (str[i] >= '0' && str[i] <= '9') {
            ans += str[i];
        } else {
            if (ans.length() > 0) { // 获取当前个数 
                cnt = stoi(ans);
            }
             while (cnt--)
                    cout << str[i];
            cnt = 1;
            ans = "";
        }
    }
}

int main() {
    cin >> ch;
    getchar();      
    getline(cin, str);

    if (ch == 'D') {
        Decompress();
    } else if (str.length() != 0) {
        Compression();
    }

    cout << ans << endl;

    return 0;
}