problems(rs): add “1431. Kids With the Greatest Number of Candies”

Signed-off-by: Matej Focko <mfocko@redhat.com>
This commit is contained in:
Matej Focko 2023-04-17 18:51:10 +02:00
parent 43d043b5e1
commit d8009b9a4a
Signed by: mfocko
GPG key ID: 7C47D46246790496

View 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]
);
}
}