持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第24天，点击查看活动详情

力扣——1235.规划兼职工作

1235. 规划兼职工作

你打算利用空闲时间来做兼职工作赚些零花钱。

这里有 n 份兼职工作，每份工作预计从 startTime[i] 开始到 endTime[i] 结束，报酬为 profit[i]。

给你一份兼职工作表，包含开始时间 startTime，结束时间 endTime 和预计报酬 profit 三个数组，请你计算并返回可以获得的最大报酬。

注意，时间上出现重叠的 2 份工作不能同时进行。

如果你选择的工作在时间 X 结束，那么你可以立刻进行在时间 X 开始的下一份工作。

示例 1：

输入：startTime = [1,2,3,3], endTime = [3,4,5,6], profit = [50,10,40,70]
输出：120
解释：
我们选出第 1 份和第 4 份工作， 
时间范围是 [1-3]+[3-6]，共获得报酬 120 = 50 + 70。

示例 2：

输入：startTime = [1,2,3,4,6], endTime = [3,5,10,6,9], profit = [20,20,100,70,60]
输出：150
解释：
我们选择第 1，4，5 份工作。 
共获得报酬 150 = 20 + 70 + 60。

提示：

• 1 <= startTime.length == endTime.length == profit.length <= 5 * 10^4
• 1 <= startTime[i] < endTime[i] <= 10^9
• 1 <= profit[i] <= 10^4

问题解析

线性dp。

设置状态转移数组f，f[i]表示当结束时间为i时，所能获得的最大报酬。

有状态转移方程：

f[i]=max(f[i-1],f[starTime[j]]+profit[j]);

但是要注意：

1. 结束时间和开始时间的范围为1e9，所以dp前要对时间进行离散化。
2. 题目给我们的兼职的结束时间并不是有序的，我们要把他们按照结束时间进行升序排序。

AC代码

bool cmp(vector<int>&a,vector<int>&b)
{
    if(a[1]!=b[1])return a[1]<b[1];
    return a[0]<b[0];
}
class Solution {
public:
    int jobScheduling(vector<int>& startTime, vector<int>& endTime, vector<int>& profit) {
        vector<int>v;
        for(auto i:startTime)v.push_back(i);
        for(auto i:endTime)v.push_back(i);
        sort(v.begin(),v.end());
        int cnt=1;
        unordered_map<int,int>mymap;
        //离散化
        for(auto i:v)
        {
            if(!mymap.count(i))
            {
                mymap[i]=cnt++;
            }
        }
        vector<int>f(cnt);
        int n=startTime.size(),idx=0,ans=1;
        vector<vector<int>>res;
        for(int i=0;i<n;i++)
        {
            res.push_back({startTime[i],endTime[i],profit[i]});
        }
        sort(res.begin(),res.end(),cmp);
        while(idx<n)
        {
            if(mymap[res[idx][1]]==ans)
            {
                while(idx<n&&mymap[res[idx][1]]==ans)
                {
                    f[ans]=max(f[ans],max(f[ans-1],f[mymap[res[idx][0]]]+res[idx][2]));
                    idx++;
                }
            }
            else f[ans]=f[ans-1];
            ans++;
        }
        return f[ans-1];
    }
};