解题思路:
本文要求修改输入数组并返回该数组的新长度,并且只能用常量空间,那么最好的方法应该是双指针。定义两个变量,一个代表当前字符出现的第一个位置,一个表示修改后当前长度,顺序遍历整个字符数组,遇到临界点(最后一个位置,或者两字符分界处)进行操作,当前字符长度获取通过遍历到的 i - left + 1即可,利用to_string()变成字符串一个一个放入原数组中,最后更新left的位置,代码如下:
class Solution {
public:
int compress(vector<char>& chars) {
int n = chars.size();
// 双指针,一个代表当前字符出现的第一个位置,一个表示修改后当前长度
int left = 0, len = 0;
for(int i = 0; i < n; i ++) {
// 遇到临界点(最后一个位置,或者两字符分界处)
if(i == n - 1 || chars[i] != chars[i + 1]) {
chars[len ++] = chars[i];
int nums = i - left + 1;
if(nums > 1) {
// 把数字加到末尾
for(char num : to_string(nums)) {
chars[len ++] = num;
}
}
left = i + 1;
}
}
return len;
}
};