题目
字符串压缩。利用字符重复出现的次数,编写一种方法,实现基本的字符串压缩功能。比如,字符串aabcccccaaa
会变为a2b1c5a3
。若“压缩”后的字符串没有变短,则返回原先的字符串。你可以假设字符串中只包含大小写英文字母(a至z)。
示例1:
输入:"aabcccccaaa" 输出:"a2b1c5a3"
示例2:
输入:"abbccd" 输出:"abbccd" 解释:"abbccd"压缩后为"a1b2c2d1",比原字符串长度更长。
提示:
- 字符串长度在[0, 50000]范围内。
代码
char* compressString(char* S){
if(strlen(S)==0||strlen(S)==1)
return S;
else {
char* p = (char*)malloc(2 * strlen(S) * sizeof(char));
char op = S[0];
int ret = 0;
int j;
int cnt = 1;
for (int i = 1; S[i - 1] != '\0'; i++) {
if (S[i] == op) {
cnt++;
}
else {
p[ret] = op;
ret++;
if (cnt >= 1) {
ret+=sprintf(&p[ret], "%d", cnt);
}
op = S[i];
cnt = 1;
}
j = i;
}
p[ret] = op;
ret++;
if (cnt > 1) {
ret+=sprintf(&p[ret], "%d", cnt);
}
p[ret] = '\0';
if (j >= ret) {
return p;
}
else {
free(p);
return S;
}
}
}
收获
1.在力扣刷题时需要返回数组的话都是返回指针的形式,因此需要使用malloc函数
2.在使用malloc后需要释放内存,如果调用的内存没有使用(也就是没有return这块内存),则需要当场free掉这块内存,如果这块内存需要被使用则不需要在这段函数里面free,相当于把这块内存调用走了,而谁调用走谁就有责任把它free。
3.明白了sprintf的使用规则
即包含头文件#linclude<string.h>使用时可以把任何形式的类型都格式换输入到字符串数组中
eg.sprintf(&p[i],"%d%lf%c",a,b,c)其中a是int类型,b是double类型,c是char类型。