20240119 Finished

This commit is contained in:
jackfiled 2024-01-19 12:16:20 +08:00
parent 9b83dfbdb2
commit 7e49f1316d
2 changed files with 67 additions and 1 deletions

View File

@ -20,4 +20,5 @@ mod p83_remove_duplicates_from_sorted_list;
mod p82_remove_duplicates_from_sorted_list_ii;
mod p2719_count_of_integers;
mod p2744_find_maximum_number_of_string_pairs;
mod p2171_removing_minimum_number_of_magic_beans;
mod p2171_removing_minimum_number_of_magic_beans;
mod p2809_minimum_time_to_make_array_sum_at_most_x;

View File

@ -0,0 +1,65 @@
/**
* [2809] Minimum Time to Make Array Sum At Most x
*/
pub struct Solution {}
// submission codes start here
struct Pair {
num1: i32,
num2: i32,
}
use std::cmp::max;
impl Solution {
pub fn minimum_time(nums1: Vec<i32>, nums2: Vec<i32>, x: i32) -> i32 {
let mut pairs = Vec::with_capacity(nums1.len());
let sum1: i32 = nums1.iter().sum();
let sum2: i32 = nums2.iter().sum();
for (index, value) in nums1.iter().enumerate() {
pairs.push(Pair {
num1: *value,
num2: nums2[index],
});
}
pairs.sort_by(|a, b| a.num2.cmp(&b.num2));
let mut dp = vec![0; nums1.len() + 1];
for i in 1..=nums1.len() {
let (num1, num2) = (pairs[i - 1].num1, pairs[i - 1].num2);
for j in (1..=i).rev() {
dp[j] = max(dp[j], dp[j - 1] + num2 * j as i32 + num1);
}
}
for i in 0..=nums1.len() {
let j = i as i32;
if sum1 + sum2 * j - dp[i] <= x {
return j;
}
}
-1
}
}
// submission codes end
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_2809() {
assert_eq!(Solution::minimum_time(vec![1, 2, 3], vec![1, 2, 3], 4),
3);
assert_eq!(Solution::minimum_time(vec![1, 2, 3], vec![3, 3, 3], 4),
-1);
assert_eq!(Solution::minimum_time(vec![4, 4, 9, 10], vec![4, 4, 1, 3], 16),
4);
}
}