leetcode - 1235. Maximum Profit in Job Scheduling

KpLn_HJL

已于 2024-01-06 13:14:51 修改

阅读量266

点赞数

分类专栏： OJ题目记录文章标签： leetcode 算法职场和发展

于 2023-08-26 11:50:46 首次发布

本文链接：https://blog.csdn.net/sinat_41679123/article/details/132509883

版权

OJ题目记录专栏收录该内容

515 篇文章 1 订阅

订阅专栏

该文章讨论了如何在给定的一组任务中，每个任务有开始时间和结束时间，以及利润，找到一个子集，使得这些任务不重叠且总利润最大。使用动态规划和二分查找算法解决此问题，时间复杂度为O(nlogn)，空间复杂度为O(n)。

摘要由CSDN通过智能技术生成

Description

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 * 10^4
1 <= startTime[i] < endTime[i] <= 10^9
1 <= profit[i] <= 10^4

Solution

Same as 2830. Maximize the Profit as the Salesman, this one is easier to read.

Use dp[i] to denote the maximum profit we could get by scheduling jobs[:i+1], then transformation equation is:
$\max (dp[i - 1], dp[j] + profit[i])$
where j is the latest job that is earlier than i

We could sort pairs by the end to use binary search, so the time complexity for searching would be $\log n$
Time complexity: $\log n)$
Space complexity: $o (n)$

Code

class Solution:
    def jobScheduling(self, startTime: List[int], endTime: List[int], profit: List[int]) -> int:
        def find_latest_interval(nums: list, target: int) -> int:
            left, right = 0, len(nums) - 1
            while left < right:
                mid = (left + right + 1) >> 1
                if nums[mid][0] > target:
                    right = mid - 1
                else:
                    left = mid
            return (left + right) >> 1

        intervals = [(startTime[i], endTime[i], profit[i]) for i in range(len(profit))]
        intervals.sort(key=lambda x: (x[1], -x[2], -x[0]))
        # (endTime, profit)
        dp = [(-1, 0)]
        for i in range(len(intervals)):
            st, ed, p = intervals[i]
            index = find_latest_interval(dp, st)
            if p + dp[index][1] > dp[-1][1]:
                dp.append((ed, dp[index][1] + p))
        return dp[-1][1]