diff options
Diffstat (limited to 'group-the-people-given-the-group-size-they-belong-to/src')
| -rw-r--r-- | group-the-people-given-the-group-size-they-belong-to/src/main.rs | 26 | 
1 files changed, 26 insertions, 0 deletions
diff --git a/group-the-people-given-the-group-size-they-belong-to/src/main.rs b/group-the-people-given-the-group-size-they-belong-to/src/main.rs new file mode 100644 index 0000000..f380f39 --- /dev/null +++ b/group-the-people-given-the-group-size-they-belong-to/src/main.rs @@ -0,0 +1,26 @@ +fn main() { +    println!("Hello, world!"); +} + +struct Solution {} +impl Solution { +    pub fn group_the_people(group_sizes: Vec<i32>) -> Vec<Vec<i32>> { +        let mut result: Vec<Vec<i32>> = Vec::new(); + + +        for (i, size) in group_sizes.iter().enumerate() { +            let group = result.iter_mut() +                .find(|x| x.capacity() == *size as usize && x.len() != x.capacity()); + +            match group { +                Some(entry) => entry.push(i as i32), +                None => { +                    let mut new_group = Vec::with_capacity(*size as usize); +                    new_group.push(i as i32); +                    result.push(new_group); +                } +            }; +        } +        result +    } +}  | 
