面试题01.06字符串压缩

题目描述

字符串压缩。利用字符重复出现的次数，编写一种方法，实现基本的字符串压缩功能。比如，字符串aabcccccaaa会变为a2b1c5a3。若“压缩”后的字符串没有变短，则返回原先的字符串。你可以假设字符串中只包含大小写英文字母（a至z）。

示例1:
输入：“aabcccccaaa”
输出：“a2b1c5a3”

示例2:
输入：“abbccd”
输出：“abbccd”
解释：“abbccd"压缩后为"a1b2c2d1”，比原字符串长度更长。
提示：
字符串长度在[0, 50000]范围内。

char* compressString(char* S){
    int len=strlen(S);
    if(len==0)
        return "";
    char* str=(char*)malloc(2*len+1);
    char ch=S[0];
    int cnt=1;
    int idx=0;
    for(int i=1;i<len;i++){
        if(S[i]==ch){
            cnt++;
        }else{
            if(cnt >= 10) {
                str[idx++] = ch;
                sprintf(str + idx, "%d", cnt);
                idx += strlen(str + idx);
                ch = S[i];
                cnt = 1;
            }else{
            str[idx++]=ch;
            str[idx++]=cnt+'0';
            ch=S[i];
            cnt=1;
            }
        }
    }
      if(cnt >= 10) {
        str[idx++] = ch;
        sprintf(str + idx, "%d", cnt);
        idx += strlen(str + idx);
      }else{
     str[idx++]=ch;
     str[idx++]=cnt+'0';
     str[idx++]='\0';
      }
    return strlen(str)<len?str:S;
}