压缩字符串
2020-04-06 本文已影响0人
MrHitchcock
-
题目描述:给定一组字符,使用原地算法将其压缩。
压缩后的长度必须始终小于或等于原数组长度。
数组的每个元素应该是长度为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。
- 解题思路:
- 使用读写双指针 read & write
- 添加 anchor 指针 代表连续字符串的起始位置
- 当read指针 + 1 等于字符串长度 或 read指针 + 1不指向同一字符时,开始write
4.write指针写入内容包括当前anchor指针指向的字符,以及(read - anchor + 1)的字符长度
- Python版:
class Solution(object):
def compress(self, chars):
anchor = write = 0
for read, c in enumerate(chars):
if read + 1 == len(chars) or chars[read + 1] != c:
chars[write] = chars[anchor]
write += 1
if read > anchor:
for digit in str(read - anchor + 1):
chars[write] = digit
write += 1
anchor = read + 1
return write
-
Tips:每次遍历结束后,anchor指针需重新指向新的连续字符起始处
-
复杂度分析
- 时间复杂度:O(N)O(N),其中 NN 是 chars 的长度。
- 空间复杂度:O(1)O(1),三个指针的占用空间。
-
Java版:
class Solution {
public int compress(char[] chars) {
int anchor = 0, write = 0;
for (int read = 0; read < chars.length; read++) {
if (read + 1 == chars.length || chars[read + 1] != chars[read]) {
chars[write++] = chars[anchor];
if (read > anchor) {
for (char c: ("" + (read - anchor + 1)).toCharArray()) {
chars[write++] = c;
}
}
anchor = read + 1;
}
}
return write;
}
}