sfcpl 0.1.1

private library for atcoder
Documentation 
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

use cargo_snippet::snippet;

/// ランレングス圧縮
#[snippet("run_length_encoding")]
pub fn run_length_encoding<T: PartialEq + Copy>(v: &Vec<T>) -> Vec<(T, usize)> {
    let mut res = Vec::new();
    let mut c = 1usize;
    let mut h = v[0];
    for e in v.iter().skip(1) {
        if h != *e {
            res.push((h, c));
            c = 1;
            h = *e;
        } else {
            c += 1;
        }
    }
    res.push((h, c));

    res
}

#[test]
fn rle_test() {
    let v = "abbccc";
    let e = run_length_encoding(&v.chars().collect());
    assert_eq!(e, vec![('a', 1), ('b', 2), ('c', 3)]);

    let v = vec![1, 1, 2, 2, 1, 1, 2, 2];
    let e = run_length_encoding(&v);
    assert_eq!(e, vec![(1, 2), (2, 2), (1, 2), (2, 2)]);
}