1078 字符串压缩与解压 (20分)---再做

#include<stdio.h>
#include<iostream>
using namespace std;
int main(){
    char a;
    string str;
    scanf("%c",&a);
    getchar();
    getline(cin,str);
    if(a=='C'){
        for(int i=0;i<str.size();i++){
            int cnt=0;
            while(str[i]==str[i+1]){
                cnt++;
                i++;
            }
            if(cnt!=0) printf("%d",cnt+1);
            printf("%c",str[i]);
        }
    }
    if(a=='D'){
        for(int i=0;i<str.size();i++){
            int cnt=0;
            while(str[i]>='0'&&str[i]<='9'){
                cnt=cnt*10+str[i]-'0';
                i++;
            }
            if(cnt==0) printf("%c",str[i]);
            else{
                for(int j=0;j<cnt;j++){
                    printf("%c",str[i]);
                }
            }
        }
    }
    return 0;
}

第一次，参考博客完成，40min