PAT Basic Level 1078 字符串压缩与解压 （20 分）

题目链接:

https://pintia.cn/problem-sets/994805260223102976/problems/994805262018265088

AC代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
using namespace std;
int main(){
    char t;
    scanf("%c",&t);
    getchar();
    string s;
    getline(cin,s);
    int cnt=1;
    if(t=='D'){
        string num="";
        for(int i=0;i<s.size();i++){
            if(s[i]>='0'&&s[i]<='9')
                num+=s[i];
            else{
                if(num.size()>0)
                    cnt=atoi(num.c_str());
                while(cnt--)
                    cout<<s[i];
                num="";
                cnt=1;
            }
        }
    }
    else if(s.size()>0){
        char pre=s[0];
        for(int i=1;i<s.size();i++){
            if(pre==s[i])
                cnt++;
            else{
                if(cnt>=2)
                    printf("%d",cnt);
                cout<<pre;
                pre=s[i];
                cnt=1;
            }
        }
        if(cnt>=2)
            cout<<cnt;
        cout<<pre;
    }
    return 0;
}

二刷:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
using namespace std;

int main(){
    char c;
    string str;
    scanf("%c",&c);
    getchar();
    if(c=='C'){
        getline(cin,str);
        char pre=str[0];
        int cnt=0;
        for(int i=1;i<str.size();i++){
            if(str[i]==pre)
                cnt++;
            else{
                if(cnt>=1)
                printf("%d",cnt+1);
                printf("%c",pre);
                cnt=0;
            }
            pre=str[i];
        }
        if(cnt>=1)
            printf("%d",cnt+1);
        printf("%c",pre);
    }
    else if(c=='D'){
        getline(cin,str);
        int cnt=1;//格式错误的话，请检查解压的时候，单个字符是否正确。例如：输入D a,输出a。测试点2就是单个字符的检测
        string num="";
        for(int i=0;i<str.size();i++){
            if(str[i]>='0'&&str[i]<='9')
                num+=str[i];
            else{
                if(num.size()>0)
                    cnt=atoi(num.c_str());
                while(cnt--)
                    cout<<str[i];
                num="";
                cnt=1;
            }
        }
    }