dsalgo 0.3.10

A package for Datastructures and Algorithms.
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

pub fn labels_to_groups(labels: &[usize]) -> Vec<Vec<usize>> {
    let n = labels.len();

    let k = *labels.iter().max().unwrap() + 1;

    let mut groups = vec![vec![]; k];

    for i in 0..n {
        groups[labels[i]].push(i);
    }

    groups
}

#[cfg(test)]

mod tests {

    use super::*;

    #[test]

    fn test() {
        let labels = vec![0, 2, 1, 0];

        assert_eq!(
            labels_to_groups(&labels),
            vec![vec![0, 3], vec![2], vec![1]]
        );
    }
}