1235. Maximum Profit in Job Scheduling
We have n
jobs, where every job is scheduled to be done from startTime[i]
to endTime[i]
, obtaining a profit of profit[i]
.
You're given the startTime
, endTime
and profit
arrays, return the maximum profit you can take such that there are no two jobs in the subset with overlapping time range.
If you choose a job that ends at time X
you will be able to start another job that starts at time X
.
Example 1:
Input: startTime = [1,2,3,3], endTime = [3,4,5,6], profit = [50,10,40,70] Output: 120 Explanation: The subset chosen is the first and fourth job. Time range [1-3]+[3-6] , we get profit of 120 = 50 + 70.
Example 2:
Input: startTime = [1,2,3,4,6], endTime = [3,5,10,6,9], profit = [20,20,100,70,60] Output: 150 Explanation: The subset chosen is the first, fourth and fifth job. Profit obtained 150 = 20 + 70 + 60.
Example 3:
Input: startTime = [1,1,1], endTime = [2,3,4], profit = [5,6,4] Output: 6
Constraints:
1 <= startTime.length == endTime.length == profit.length <= 5 * 104
1 <= startTime[i] < endTime[i] <= 109
1 <= profit[i] <= 104
Rust Solution
struct Solution;
use std::collections::BTreeMap;
impl Solution {
fn job_scheduling(start_time: Vec<i32>, end_time: Vec<i32>, profit: Vec<i32>) -> i32 {
let mut jobs = vec![];
let n = start_time.len();
for i in 0..n {
jobs.push((start_time[i], end_time[i], profit[i]));
}
jobs.sort_unstable();
let mut memo: BTreeMap<i32, i32> = BTreeMap::new();
let mut res = 0;
for i in (0..n).rev() {
let mut cur = jobs[i].2;
if let Some((_, val)) = memo.range(jobs[i].1..).next() {
cur += val;
}
res = res.max(cur);
memo.insert(jobs[i].0, res);
}
res
}
}
#[test]
fn test() {
let start_time = vec![1, 2, 3, 3];
let end_time = vec![3, 4, 5, 6];
let profit = vec![50, 10, 40, 70];
let res = 120;
assert_eq!(Solution::job_scheduling(start_time, end_time, profit), res);
let start_time = vec![1, 2, 3, 4, 6];
let end_time = vec![3, 5, 10, 6, 9];
let profit = vec![20, 20, 100, 70, 60];
let res = 150;
assert_eq!(Solution::job_scheduling(start_time, end_time, profit), res);
let start_time = vec![1, 1, 1];
let end_time = vec![2, 3, 4];
let profit = vec![5, 6, 4];
let res = 6;
assert_eq!(Solution::job_scheduling(start_time, end_time, profit), res);
}
Having problems with this solution? Click here to submit an issue on github.