LeetCode 443 压缩字符串[双指针] HERODING的LeetCode之路

Posted 2021-09-07 HERODING23

解题思路：
本文要求修改输入数组并返回该数组的新长度，并且只能用常量空间，那么最好的方法应该是双指针。定义两个变量，一个代表当前字符出现的第一个位置，一个表示修改后当前长度，顺序遍历整个字符数组，遇到临界点（最后一个位置，或者两字符分界处）进行操作，当前字符长度获取通过遍历到的 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;
    }
};