From 6360f90e9e5eb78fffa4485bba721e7798a12eea Mon Sep 17 00:00:00 2001 From: jackfiled Date: Fri, 6 Sep 2024 15:57:51 +0800 Subject: [PATCH] 20240906 finished. --- src/problem/mod.rs | 3 +- ..._maximum_length_of_a_good_subsequence_i.rs | 41 +++++++++++++++++++ 2 files changed, 43 insertions(+), 1 deletion(-) create mode 100644 src/problem/p3176_find_the_maximum_length_of_a_good_subsequence_i.rs diff --git a/src/problem/mod.rs b/src/problem/mod.rs index 88ea177..de09f06 100644 --- a/src/problem/mod.rs +++ b/src/problem/mod.rs @@ -226,4 +226,5 @@ mod p1450_number_of_students_doing_homework_at_a_given_time; mod p2024_maximize_the_confusion_of_an_exam; mod p2708_maximum_strength_of_a_group; mod p2860_happy_students; -mod p3174_clear_digits; \ No newline at end of file +mod p3174_clear_digits; +mod p3176_find_the_maximum_length_of_a_good_subsequence_i; \ No newline at end of file diff --git a/src/problem/p3176_find_the_maximum_length_of_a_good_subsequence_i.rs b/src/problem/p3176_find_the_maximum_length_of_a_good_subsequence_i.rs new file mode 100644 index 0000000..2a39805 --- /dev/null +++ b/src/problem/p3176_find_the_maximum_length_of_a_good_subsequence_i.rs @@ -0,0 +1,41 @@ +/** + * [3176] Find the Maximum Length of a Good Subsequence I + */ +pub struct Solution {} + + +// submission codes start here + +impl Solution { + pub fn maximum_length(nums: Vec, k: i32) -> i32 { + use std::collections::HashMap; + + let k = k as usize; + let mut map = HashMap::with_capacity(nums.len()); + let mut mx = vec![0; k + 2]; + + for x in nums { + let entry = map.entry(x).or_insert(vec![0; k + 1]); + + for j in (0..=k).rev() { + entry[j] = entry[j].max(mx[j]) + 1; + mx[j + 1] = mx[j + 1].max(entry[j]); + } + } + + mx[k + 1] + } +} + +// submission codes end + +#[cfg(test)] +mod tests { + use super::*; + + #[test] + fn test_3176() { + assert_eq!(4, Solution::maximum_length(vec![1, 2, 1, 1, 3], 2)); + assert_eq!(2, Solution::maximum_length(vec![1,2,3,4,5,1], 0)); + } +}