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 + } +} |