Leetcode 面试题 01.06.字符串压缩（C语言）

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

示例1

 输入："aabcccccaaa"
 输出："a2b1c5a3"

示例2

输入："abbccd"
 输出："abbccd"
 解释："abbccd"压缩后为"a1b2c2d1"，比原字符串长度更长。

提示：

字符串长度在[0, 50000]范围内。

 

代码：

char* compressString(char* S)
{
    int length = strlen(S);
    char num_arr[5] = {0};
    int length_num;
    char *ans = (char *)malloc(sizeof(char) * 60000);
    int i = 0;
    int j = 0;
    int k;
    int num = 0;

    while(S[i] != 0)
    {
        if(S[i] == S[i + 1])
        {
            num++;
        }
        else
        {
            ans[j++] = S[i];
            sprintf(num_arr,"%d",num + 1);
            length_num = strlen(num_arr);
            for(k = 0; k < length_num; k++)
            {
                ans[j++] = num_arr[k];
            }
            num = 0;
        }
        i++;
    }
    ans[j] = 0;

    if(j >= length)
    {
        return S;
    }
    else
    {
        return ans;
    }
}