利用双指针法:
为了实现原地压缩,我们可以使用双指针分别标志我们在字符串中读read
和写wrute
的位置。每次当读指针 read
移动到某一段连续相同子串的最右侧,我们就在写指针 write
处依次写入该子串对应的字符和子串长度即可。
在实际代码中,当读指针 read
位于字符串的末尾,或读指针 read
指向的字符不同于下一个字符时,我们就认为读指针 read
位于某一段连续相同子串的最右侧。该子串对应的字符即为读指针 read
指向的字符串。我们使用变量 left
记录该子串的最左侧的位置,这样子串长度即为 read−left+1
。
特别地,为了达到 O(1)空间复杂度,我们需要自行实现将数字转化为字符串写入到原字符串的功能。这里我们采用短除法
将子串长度倒序
写入原字符串中,然后再将其反转
即可。
class Solution {
public:
int compress(vector<char>& chars) {
int n = chars.size();
int write = 0, left = 0;
//遍历整个字符数组
for (int read = 0; read < n; read++) {
//如果当前为末尾字符或下一个字符与当前字符不一致
if (read == n - 1 || chars[read] != chars[read + 1]) {
chars[write++] = chars[read]; //将当前字符写入write位置处
int num = read - left + 1; //计算重复字符的个数
if (num > 1) { //如果字符个数大于1
int anchor = write; //先暂存写指针位置,方便后续反转
while (num > 0) { //利用短除法将字符个数转为字符(注意这里是倒叙的)
chars[write++] = num % 10 + '0';
num /= 10;
}
reverse(&chars[anchor], &chars[write]); //再将字符个数的字符表示翻转
}
left = read + 1; //更新left指针
}
}
return write; //write即为新数组的长度
}
};