Basic Level 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

思路分析：

1. 解压时候，另count=1如果遇到数字则将字符串转化为int类型，然后输出count个当前字符
2. 压缩时候，用temp保存前一个字符，如果与当前字符相同则count++，否则输出count和temp。最后也得输出一次。

代码：

#include <bits/stdc++.h>
using namespace std;
int main(){
    char op, temp;
    cin >> op;
    getchar();
    string s, num;
    getline(cin, s);
    int count = 1;
    if(op == 'C'){
        temp = s[0];
        for(int i = 1; i < s.length(); i++)
            if(s[i] == temp) count++;
            else{
                if(count > 1) cout << count;
                cout << temp;
                count = 1;
                temp = s[i];
             }
        if(count > 1) cout << count;
        cout << temp;
    }else if(op == 'D'){
        for(int i = 0; i < s.length(); i++){
            if(s[i] >= '0' && s[i] <= '9'){
                num += s[i];
            }else{
                if(num.length() > 0) count = stoi(num);
                for(int j = 0; j < count; j++) cout << s[i];
                count = 1;
                num = "";
            }
        }
    }
    
}

---

PAT_BasicLevel