leetcode/algorithms/StringCompression/solution.js at master · P-ppc/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
/**
 * @param {character[]} chars
 * @return {number}
 */
var compress = function(chars) {
    let char = null,
        count = 0,
        length = 0

    chars.forEach(c => {
        if (char === null) {
            count = 1
            char = c
        } else if (char === c) {
            count++
        } else if (char !== c) {
            chars[length] = char
            length++

            if (count > 1) {
                let countStr = String(count)
                for (let j = 0; j < countStr.length; j++) {
                    chars[length] = countStr[j]
                    length++
                }
            }

            count = 1
            char = c
        }
    })

    chars[length] = char
    length++

    if (count > 1) {
        let countStr = String(count)
        for (let j = 0; j < countStr.length; j++) {
            chars[length] = countStr[j]
            length++
        }
    }

    return length
};