20240119 Finished
This commit is contained in:
parent
9b83dfbdb2
commit
7e49f1316d
|
@ -21,3 +21,4 @@ mod p82_remove_duplicates_from_sorted_list_ii;
|
||||||
mod p2719_count_of_integers;
|
mod p2719_count_of_integers;
|
||||||
mod p2744_find_maximum_number_of_string_pairs;
|
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;
|
|
@ -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);
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user