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

点击链接PAT乙级-AC全解汇总

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

我的代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<string>
#include<set>
#include<map>
#include<algorithm>
#include<cmath>
#include<ctime>
#include<cstring>
#include<sstream>
using namespace std;
//有的时候题目是一起做的，所以会有不需要的头文件

int main()
{
    char c;
    scanf("%c",&c);
    getchar();//如果是scanf("%c\n",&c)case3会报错
    string src;
    getline(cin,src);
    if(c=='C')//压缩
    {
        for(int i=0;i<src.size();i++)
        {
            if(src[i]==src[i+1])
            {
                int j=i+1;
                while(j<src.length()&&src[j]==src[i])j++;
                int num=j-i;
                string temp=to_string(num);
                temp+=src[i];
                src.replace(i,j-i,temp);//所有相同字母替换
            }
        }
    }
    else //解压
    {
        if(src.length()==0)cout<<" ";//case 2
        else
        for(int i=0;i<src.size();i++)//解压的时候遍历完整
        {
           if(src[i]>='0'&&src[i]<='9')
           {
               int j=i+1;
               while(j<src.length()&&src[j]>='0'&&src[j]<='9')j++;
               int num=stoi(src.substr(i,j-i));
               string temp;
               while(num-->0)temp+=src[j];
               src.replace(i,j-i+1,temp);//数字和后一个字母替换
           }
        }
    }
    cout<<src;
    return 0;
}

注意：如果是scanf("%c\n",&c)case3会报错，我也不知道为什么，必须要用getchar()；
注意：case 2解压输出一个空格的情况；