problems(rs): add “1431. Kids With the Greatest Number of Candies”
Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
parent
43d043b5e1
commit
d8009b9a4a
1 changed files with 41 additions and 0 deletions
41
problems/kids-with-the-greatest-number-of-candies.rs
Normal file
41
problems/kids-with-the-greatest-number-of-candies.rs
Normal file
|
@ -0,0 +1,41 @@
|
|||
struct Solution {}
|
||||
impl Solution {
|
||||
pub fn kids_with_candies(candies: Vec<i32>, extra_candies: i32) -> Vec<bool> {
|
||||
let max = *candies.iter().max().unwrap();
|
||||
candies
|
||||
.iter()
|
||||
.map(|candies| candies + extra_candies >= max)
|
||||
.collect()
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn example_1() {
|
||||
assert_eq!(
|
||||
Solution::kids_with_candies(vec![2, 3, 5, 1, 3], 3),
|
||||
vec![true, true, true, false, true]
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn example_2() {
|
||||
assert_eq!(
|
||||
Solution::kids_with_candies(vec![4, 2, 1, 1, 2], 1),
|
||||
vec![true, false, false, false, false]
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn example_3() {
|
||||
assert_eq!(
|
||||
Solution::kids_with_candies(vec![12, 1, 12], 10),
|
||||
vec![true, false, true]
|
||||
);
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue