443. String Compression
Tags
- Two Pointers
- String
Link
Question
Given an array of characters
chars
, compress it using the following algorithm:
Begin with an empty strings
. For each group of consecutive repeating characters inchars
:
- If the group's length is
1
, append the character tos
.- Otherwise, append the character followed by the group's length.
The compressed string
s
should not be returned separately, but instead, be stored in the input character arraychars
. Note that group lengths that are10
or longer will be split into multiple characters inchars
.
After you are done modifying the input array, return the new length of the array.
You must write an algorithm that uses only constant extra space.
Example 1:
Input: chars = ["a","a","b","b","c","c","c"]
Output: Return 6, and the first 6 characters of the input array should be: ["a","2","b","2","c","3"]
Explanation: The groups are "aa", "bb", and "ccc". This compresses to "a2b2c3".
Example 2:
Input: chars = ["a"]
Output: Return 1, and the first character of the input array should be: ["a"]
Explanation: The only group is "a", which remains uncompressed since it's a single character.
Example 3:
Input: chars = ["a","b","b","b","b","b","b","b","b","b","b","b","b"]
Output: Return 4, and the first 4 characters of the input array should be: ["a","b","1","2"].
Explanation: The groups are "a" and "bbbbbbbbbbbb". This compresses to "ab12".
Constraints:
1 <= chars.length <= 2000
chars[i]
is a lowercase English letter, uppercase English letter, digit, or symbol.
Answer
JavaScript
/**
* @param {character[]} chars
* @return {number}
*/
var compress = function (chars) {
let writeIndex = 0;
let readIndex = 0;
while (readIndex < chars.length) {
let currentChar = chars[readIndex];
let count = 0;
while (readIndex < chars.length && chars[readIndex] === currentChar) {
readIndex++;
count++;
}
chars[writeIndex++] = currentChar;
if (count > 1) {
for (let digit of String(count)) {
chars[writeIndex++] = digit;
}
}
}
return writeIndex;
};
/**
* @param {character[]} chars
* @return {number}
*/
var compress = function (chars) {
let s = "";
let len = 1;
for (let i = 0; i < chars.length; i++) {
if (chars[i] === chars[i + 1]) {
len++;
} else {
if (len === 1) {
s = s + String(chars[i]);
} else {
s = s + String(chars[i]) + String(len);
len = 1;
}
}
}
chars.unshift(...s.split(""));
return s.split("").length;
};