diff --git a/src/problem/mod.rs b/src/problem/mod.rs index aa694cc..ef7b61b 100644 --- a/src/problem/mod.rs +++ b/src/problem/mod.rs @@ -366,3 +366,5 @@ mod p935_knight_dialer; mod p2717_semi_ordered_permutation; mod p2931_maximum_spending_after_buying_items; + +mod p3264_final_array_state_after_k_multiplication_operations_i; diff --git a/src/problem/p3264_final_array_state_after_k_multiplication_operations_i.rs b/src/problem/p3264_final_array_state_after_k_multiplication_operations_i.rs new file mode 100644 index 0000000..963e00c --- /dev/null +++ b/src/problem/p3264_final_array_state_after_k_multiplication_operations_i.rs @@ -0,0 +1,35 @@ +/** + * [3264] Final Array State After K Multiplication Operations I + */ +pub struct Solution {} + +// submission codes start here + +impl Solution { + pub fn get_final_state(nums: Vec, k: i32, multiplier: i32) -> Vec { + let mut nums = nums; + + for _ in 0..k { + let min_value = nums.iter_mut().min().unwrap(); + *min_value = *min_value * multiplier; + } + + nums + } +} + +// submission codes end + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_3264() { + assert_eq!( + vec![8, 4, 6, 5, 6], + Solution::get_final_state(vec![2, 1, 3, 5, 6], 5, 2) + ); + assert_eq!(vec![16, 8], Solution::get_final_state(vec![1, 2], 3, 4)); + } +}