20241231 finished.
This commit is contained in:
parent
4ce91753b4
commit
2aa12ebd2d
|
@ -402,3 +402,5 @@ mod p3046_split_the_array;
|
||||||
mod p1366_rank_teams_by_votes;
|
mod p1366_rank_teams_by_votes;
|
||||||
|
|
||||||
mod p1367_linked_list_in_binary_tree;
|
mod p1367_linked_list_in_binary_tree;
|
||||||
|
|
||||||
|
mod p3219_minimum_cost_for_cutting_cake_ii;
|
||||||
|
|
44
src/problem/p3219_minimum_cost_for_cutting_cake_ii.rs
Normal file
44
src/problem/p3219_minimum_cost_for_cutting_cake_ii.rs
Normal file
|
@ -0,0 +1,44 @@
|
||||||
|
/**
|
||||||
|
* [3219] Minimum Cost for Cutting Cake II
|
||||||
|
*/
|
||||||
|
pub struct Solution {}
|
||||||
|
|
||||||
|
// submission codes start here
|
||||||
|
|
||||||
|
impl Solution {
|
||||||
|
pub fn minimum_cost(m: i32, n: i32, horizontal_cut: Vec<i32>, vertical_cut: Vec<i32>) -> i64 {
|
||||||
|
let (m, n) = (m as usize, n as usize);
|
||||||
|
let mut horizontal_cut: Vec<i64> = horizontal_cut.into_iter().map(|x| x as i64).collect();
|
||||||
|
horizontal_cut.sort_unstable();
|
||||||
|
let mut vertical_cut: Vec<i64> = vertical_cut.into_iter().map(|x| x as i64).collect();
|
||||||
|
vertical_cut.sort_unstable();
|
||||||
|
|
||||||
|
let mut result = 0;
|
||||||
|
let (mut i, mut j) = (0, 0);
|
||||||
|
|
||||||
|
for _ in 0..m + n - 2 {
|
||||||
|
if j == n - 1 || i < m - 1 && horizontal_cut[i] < vertical_cut[j] {
|
||||||
|
result += horizontal_cut[i] * (n - j) as i64;
|
||||||
|
i += 1;
|
||||||
|
} else {
|
||||||
|
result += vertical_cut[j] * (m - i) as i64;
|
||||||
|
j += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// submission codes end
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_3219() {
|
||||||
|
assert_eq!(13, Solution::minimum_cost(3, 2, vec![1, 3], vec![5]));
|
||||||
|
assert_eq!(15, Solution::minimum_cost(2, 2, vec![7], vec![4]));
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user