https://leetcode-cn.com/problems/string-compression/
给定一组字符,使用原地算法将其压缩。
压缩后的长度必须始终小于或等于原数组长度。
数组的每个元素应该是长度为1 的字符(不是 int 整数类型)。
在完成原地修改输入数组后,返回数组的新长度。
进阶:
你能否仅使用O(1) 空间解决问题?
示例 1:
输入:
[“a”,“a”,“b”,“b”,“c”,“c”,“c”]
输出:
返回6,输入数组的前6个字符应该是:[“a”,“2”,“b”,“2”,“c”,“3”]
说明:
"aa"被"a2"替代。"bb"被"b2"替代。"ccc"被"c3"替代。
示例 2:
输入:
[“a”]
输出:
返回1,输入数组的前1个字符应该是:[“a”]
说明:
没有任何字符串被替代。
示例 3:
输入:
[“a”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”,“b”]
输出:
返回4,输入数组的前4个字符应该是:[“a”,“b”,“1”,“2”]。
说明:
由于字符"a"不重复,所以不会被压缩。"bbbbbbbbbbbb"被“b12”替代。
注意每个数字在数组中都有它自己的位置。
注意:
所有字符都有一个ASCII值在[35, 126]区间内。
1 <= len(chars) <= 1000。
思路:见程序注释
void reserve(char *s, int slen){
int i = 0;
for(; i < slen / 2; i++){
char tmp = s[i];
s[i] = s[slen - i - 1];
s[slen - i - 1] = tmp;
}
}
int compress(char* chars, int charsSize) {
// check chars
if(NULL == chars){
return 0;
}
// prepare 双指针遍历字符串,s指向每个字符串头部,p计算重复位数
char *s = chars;
char *p = s;
int slen = 0;
int clen = 0;
// while
while(clen != charsSize){
s = p;
chars[slen++] = *s;
int count_p = 0;
while(*p == *s){ // 计算重复位数
count_p++;
clen++;
if(clen == charsSize){ // 末尾处理
goto end;
}
p++;
}
end:
if(1 == count_p){ // 重复位数为1, 直接放字母
continue;
}else{
int reslen = 0; // 位数
char *tmp = &chars[slen];
while(count_p != 0){
int num = count_p % 10;
count_p /= 10;
reslen++;
chars[slen++] = num + '0';
}
reserve(tmp, reslen); //重复数 》 1 , 逆置
}
}
return slen;
}