Facebook Pixel

Minimum Swaps to Group All 1's Together

Medium
LeetCode ↗

Given an array of characters chars, compress it using the following algorithm:

Begin with an empty string s. For each group of consecutive repeating characters in chars:

  • If the group's length is 1, append the character to s.
  • 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 array chars. Note that group lengths that are 10 or longer will be split into multiple characters in chars.

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.

Explanation

Use two pointers moving in the same direction: read scans the original array, and write builds the compressed result in place. The key idea is to process one run of equal characters at a time. While read stays on the same character, we increase a count. When the run ends (or we reach the end of the array), we write that character at chars[write], then move write forward.

If count == 1, we stop there. If count > 1, we convert the count to a string (for example, 12 -> "12") and write each digit into chars, advancing write after each digit. Then we start counting the next run.

Because each character is read once and written at most once (plus count digits), this is linear time. We only use a few variables besides the input array, so extra space is constant.

Implementation

def compress(self, chars: List[str]) -> int:
    def compress_char(write, curr, counter):
        chars[write] = curr
        write += 1
        if counter == 1:      # does not append length
            return write
        length = str(counter) # convert length to string
        for c in length:
            chars[write] = c
            write += 1
        return write
    write, counter, curr = 0, 1, chars[0]
    for read in range(1, len(chars)):
        if chars[read] == curr:
            counter += 1
        else:
            write = compress_char(write, curr, counter)
            counter = 1
        curr = chars[read]
    write = compress_char(write, curr, counter)
    return write

Ready to land your dream job?

Unlock your dream job with a 5-minute evaluator for a personalized learning plan!

Start Evaluator
Discover Your Strengths and Weaknesses: Take Our 5-Minute Quiz to Tailor Your Study Plan:

What's the output of running the following function using the following tree as input?

1def serialize(root):
2    res = []
3    def dfs(root):
4        if not root:
5            res.append('x')
6            return
7        res.append(root.val)
8        dfs(root.left)
9        dfs(root.right)
10    dfs(root)
11    return ' '.join(res)
12
1import java.util.StringJoiner;
2
3public static String serialize(Node root) {
4    StringJoiner res = new StringJoiner(" ");
5    serializeDFS(root, res);
6    return res.toString();
7}
8
9private static void serializeDFS(Node root, StringJoiner result) {
10    if (root == null) {
11        result.add("x");
12        return;
13    }
14    result.add(Integer.toString(root.val));
15    serializeDFS(root.left, result);
16    serializeDFS(root.right, result);
17}
18
1function serialize(root) {
2    let res = [];
3    serialize_dfs(root, res);
4    return res.join(" ");
5}
6
7function serialize_dfs(root, res) {
8    if (!root) {
9        res.push("x");
10        return;
11    }
12    res.push(root.val);
13    serialize_dfs(root.left, res);
14    serialize_dfs(root.right, res);
15}
16

Recommended Readings

Want a Structured Path to Master System Design Too? Don’t Miss This!

Load More