【PAT乙级】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

个人思路

这题就是根据操作类型遍历待操作字符串，然后将遍历处理的结果加到结果字符串上。

本题有一个注意点：

重复字符的个数可能>=10，也就意味着有可能会有两位及以上的数字出现，因此要正确地在字符串和整型数字直接转换

建议使用sprintf()函数

代码实现

#include <cstdio>
#include <cstring>
#include <string>
#include <set>
#include <cmath>
#include <algorithm>
#include <iostream>
#define ll long long
#define ep 1e-5
#define INF 0x7FFFFFFF

using namespace std;

int main() {
    // 输入操作码
    char oper;
    cin >> oper;
    getchar();
    
    // 输入待操作字符串
    string oper_str, ans_str = "";
    getline(cin, oper_str);
    
    int len = int(oper_str.length());
    // 压缩操作
    if (oper == 'C') {
        for (int i = 0; i < len; i ++) {
            int cnt = 1;
            while (oper_str[i] == oper_str[i+1]) {
                i ++;
                cnt ++;
                if (i == len-1)
                    break;
            }
            if ((i == len-1 && oper_str[i] != oper_str[i-1]) || cnt == 1) {
                ans_str += oper_str[i];
                continue;
            }
            char cnt_str[5];
            sprintf(cnt_str, "%d", cnt); //使用sprintf函数直接将数字转化成字符串
            ans_str += cnt_str;
            ans_str += oper_str[i];
        }
    }
    // 解压缩操作
    else if (oper == 'D') {
        for (int i = 0; i < len; i ++) {
            int cnt = 0;
            while (oper_str[i] >= '0' && oper_str[i] <= '9') {
                cnt = cnt*10 + (oper_str[i]-'0');
                i ++;
            }
            if (cnt == 0) {
                cnt ++;
            }
            while (cnt --) {
                ans_str += oper_str[i];
            }
        }
    }
    
    cout << ans_str << endl;
    
    return 0;
}

总结

学习不息，继续加油