题目链接:压缩字符串
题目描述:
给定一组字符,使用原地算法将其压缩。
压缩后的长度必须始终小于或等于原数组长度。
数组的每个元素应该是长度为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” 替代。
注意每个数字在数组中都有它自己的位置。
分析:使用两个指针,一个读指针用来扫描原字符串,一个写指针用来累加压缩后的字符串的位置,如果当前重复的字符串扫描完毕之后,同时更新的压缩后的字符串的值,代表当前的这个字符之前的字符串已经全被处理过。最后返回写指针的位置就是长度。
class Solution {
public int compress(char[] chars) {
if(chars.length == 1) return 1;
int i = 0, j = 0;
while(j < chars.length) {
int count = 0;
while(j < chars.length && chars[j] == chars[i]) {
j++;
count++;
}
i++;
//每个数字都是一位字符
if(count > 1) {
char[] nums = String.valueOf(count).toCharArray();
for(int k=0; k < nums.length; k++) {
chars[i++] = nums[k];
}
}
if(j < chars.length) {
chars[i] = chars[j];//更新下一次要判断的字符
}
}
return i;
}
}
类似题:字符串压缩
不需要原地修改,简单许多
class Solution {
public String compressString(String S) {
int N = S.length();
int i = 0;
StringBuilder sb = new StringBuilder();
while (i < N) {
int j = i;
while (j < N && S.charAt(j) == S.charAt(i)) {
j++;
}
sb.append(S.charAt(i));
sb.append(j - i);
i = j;
}
String res = sb.toString();
return res.length()>=S.length()?S:res;
}
}