20241007 finished.
This commit is contained in:
parent
87428f2b3b
commit
d1610560ff
|
@ -255,4 +255,5 @@ mod p983_minimum_cost_for_tickets;
|
|||
mod p1870_minimum_speed_to_arrive_on_time;
|
||||
mod p1928_minimum_cost_to_reach_destination_in_time;
|
||||
mod p1227_airplane_seat_assignment_probability;
|
||||
mod p2187_minimum_time_to_complete_trips;
|
||||
mod p2187_minimum_time_to_complete_trips;
|
||||
mod p871_minimum_number_of_refueling_stops;
|
55
src/problem/p871_minimum_number_of_refueling_stops.rs
Normal file
55
src/problem/p871_minimum_number_of_refueling_stops.rs
Normal file
|
@ -0,0 +1,55 @@
|
|||
/**
|
||||
* [871] Minimum Number of Refueling Stops
|
||||
*/
|
||||
pub struct Solution {}
|
||||
|
||||
|
||||
// submission codes start here
|
||||
|
||||
impl Solution {
|
||||
pub fn min_refuel_stops(target: i32, start_fuel: i32, stations: Vec<Vec<i32>>) -> i32 {
|
||||
use std::collections::BinaryHeap;
|
||||
let mut stations: Vec<(i32, i32)> = stations.iter().map(|v| (v[0], v[1])).collect();
|
||||
stations.push((target, 0));
|
||||
|
||||
let mut heap = BinaryHeap::new();
|
||||
let mut fuel = start_fuel;
|
||||
let mut last_pos = 0;
|
||||
let mut count = 0;
|
||||
|
||||
for (p, f) in stations {
|
||||
while fuel < p - last_pos {
|
||||
if let Some(head) = heap.pop() {
|
||||
fuel += head;
|
||||
count += 1;
|
||||
} else {
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
|
||||
fuel -= p - last_pos;
|
||||
last_pos = p;
|
||||
heap.push(f);
|
||||
}
|
||||
|
||||
if fuel >= 0 {
|
||||
count
|
||||
} else {
|
||||
-1
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// submission codes end
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_871() {
|
||||
assert_eq!(0, Solution::min_refuel_stops(1, 1, vec![]));
|
||||
assert_eq!(-1, Solution::min_refuel_stops(100, 1, vec![vec![10, 100]]));
|
||||
assert_eq!(2, Solution::min_refuel_stops(100, 10, vec![vec![10, 60], vec![20, 30], vec![30, 30], vec![60, 40]]));
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user