rs: add «1043. Partition Array for Maximum Sum»

Signed-off-by: Matej Focko <me@mfocko.xyz>
This commit is contained in:
Matej Focko 2024-02-03 21:30:50 +01:00
parent 45fb950e25
commit f88f860d0d
Signed by: mfocko
GPG key ID: 7C47D46246790496

View file

@ -0,0 +1,21 @@
use std::cmp;
impl Solution {
pub fn max_sum_after_partitioning(arr: Vec<i32>, k: i32) -> i32 {
let mut dp: Vec<i32> = vec![0; arr.len() + 1];
for i in 1..=arr.len() {
let mut m = 0;
let mut m_sum = 0;
for j in (1..=k as usize).take_while(|j| i >= *j) {
m = cmp::max(m, arr[i - j]);
m_sum = cmp::max(m_sum, dp[i - j] + (j as i32) * m);
}
dp[i] = m_sum;
}
dp[arr.len()]
}
}