1235. 规划兼职工作

2021/02/25 leetcode 共 1063 字,约 4 分钟

title

problem link

给你一个 events 数组,其中 events[i] = [startDayi, endDayi, valuei] ,表示第 i 个会议在 startDayi 天开始,第 endDayi 天结束,如果你参加这个会议,你能得到价值 valuei 。同时给你一个整数 k 表示你能参加的最多会议数目。

你同一时间只能参加一个会议。如果你选择参加某个会议,那么你必须 完整 地参加完这个会议。会议结束日期是包含在会议内的,也就是说你不能同时参加一个开始日期与另一个结束日期相同的两个会议。

请你返回能得到的会议价值 最大和 。

示例 1:

img

输入:events = [[1,2,4],[3,4,3],[2,3,1]], k = 2
输出:7
解释:选择绿色的活动会议 0 和 1,得到总价值和为 4 + 3 = 7 。

示例 2:

img

输入:events = [[1,2,4],[3,4,3],[2,3,10]], k = 2
输出:10
解释:参加会议 2 ,得到价值和为 10 。
你没法再参加别的会议了,因为跟会议 2 有重叠。你 不 需要参加满 k 个会议。

示例 3:

img

输入:events = [[1,1,1],[2,2,2],[3,3,3],[4,4,4]], k = 3
输出:9
解释:尽管会议互不重叠,你只能参加 3 个会议,所以选择价值最大的 3 个会议。

solution

动态规划 dp(i)表示前i个,最大值,然后i可以选择参加d[i] = d[idx]+vi, idx是前面最后一个满足条件的 选择不参加,dp[i] = max(dp[i],dp[i-1])

class Solution:
    def jobScheduling(self, startTime: List[int], endTime: List[int], profit: List[int]) -> int:
        n = len(startTime)
        events = [[startTime[i], endTime[i], profit[i]] for i in range(n)]
        events = sorted(events, key=lambda x: x[1])
        ets = [x[1] for x in events]
        dp = [0 for _ in range(n + 1)]
        for i in range(1, n + 1):
            idx = 0
            st, et, vi = events[i - 1]
            # 参加
            idx = bisect.bisect_left(ets, st+1, 0, i - 1) - 1
            if idx < 0:
                dp[i] = max(dp[i], vi)
            else:
                dp[i] = max(dp[i], dp[idx + 1] + vi)
            # 不参加
            dp[i] = max(dp[i], dp[i - 1])
        return dp[n]

Search

    Table of Contents