1.题目描述
给定一组字符,使用原地算法将其压缩。压缩后的长度必须始终小于或等于原数组长度。数组的每个元素应该是长度为1 的字符(不是 int 整数类型)。在完成原地修改输入数组后,返回数组的新长度。
进阶:
你能否仅使用O(1) 空间解决问题?
示例 1:
示例 2:
示例 3:
提示:
所有字符都有一个ASCII值在[35, 126]区间内。
1 <= len(chars) <= 1000。
2.思路(双指针)
使用两个指针 read 和 write 分别标记读和写的位置。读写操作均从左到右进行:读入连续的一串字符,然后将压缩版写到数组中。最终,write 将指向输出答案的结尾。
算法:
从左到右进行读取。当读到最后一个字符,或者下一个字符与当前字符不同时,则到达连续区块的结尾。当到达连续区块的结尾时,就从 write 写入压缩的结果。chars[index] 为字符,read - index + 1 (若大于 1)为长度。
3.代码
class Solution {
public:
int compress(vector<char>& chars) {
int index = 0, write = 0;
for(int read = 0; read < chars.size(); ++read){
if(read == chars.size() - 1 || chars[read + 1] != chars[read]){
chars[write++] = chars[index];
if(read > index){ //如果相同字符长度大于1
for(auto& c : to_string(read - index + 1)){
chars[write++] = c;
}
}
index = read + 1;
}
}
return write;
}
};
4.复杂度分析
时间复杂度:O(n)
空间复杂度:O(1)