LeetCode 1235. 规划兼职工作

92 阅读3分钟

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

方法一:离散化:多次哈希 + DPx1

我们将所有出现过的时间记录下来并排序,那么我们就只需要考虑“出现过的时间”这些特殊节点,最多一共2n2n个节点。

假设一共有nn个节点(appearedTime.size()=nappearedTime.size() = n),我们建立一个长度为nndpdp数组,其中dp[i]dp[i]代表到时间appearedTime[i]appearedTime[i]为止的最大获利。

dp[i]=max{dp[i1],dp[t份工作的开始时间在appearedTime中的下标]+profit[t]}dp[i] = \max\{dp[i - 1], dp[第t份工作的开始时间在appearedTime中的下标] + profit[t]\},其中endTime[t]=appearedTime[i]endTime[t] = appearedTime[i]

什么意思呢?就是假如有一份工作在appearedTime[i]appearedTime[i]时刻结束,那么选择这份工作的话获利为这份工作开始时的最大获利+这份工作的工资=dp[这份工作开始时间对应的index]+profit[这份工作]这份工作开始时的最大获利 + 这份工作的工资 = dp[这份工作开始时间对应的index] + profit[这份工作]

以上所有需要用到的东西,均由哈希表映射即可。

如何处理出现过的时间节点

首先将所有出现过的时间放入哈希表中,然后将哈希表中的所有时间取出来,再排个序

int n = startTime.size();
// 插入哈希表
unordered_set<int> appearedTimeSet;  // 所有的出现过的时间
for (int i = 0; i < n; i++) {
    appearedTimeSet.insert(startTime[i]);
    appearedTimeSet.insert(endTime[i]);
}
// 存入数组并排序
vector<int> appearedTime;
for (const int& t : appearedTimeSet) {
    appearedTime.push_back(t);
}
sort(appearedTime.begin(), appearedTime.end());

如何由结束时间映射到这是第几份工作

<工作结束时间, 这是第几份工作>插入哈希表,就可以通过工作结束时间获取所有的在这个时间结束的工作

int n = startTime.size();

unordered_multimap<int, int> endBy;  // <在这个时间结束, 这个任务对应的编号>
for (int i = 0; i < n; i++) {
    endBy.insert({endTime[i], i});
}

如何由工作的开始时间映射到其在appearedTime中的index

遍历appearedTime中的时间,将<时间, 这个时间的index>插入哈希表

int nTime = appearedTime.size();
unordered_map<int, int> time2loc;
for (int i = 0; i < nTime; i++) {
    time2loc[appearedTime[i]] = i;
}

动态规划部分怎么实现

vector<int> dp(nTime);
for (int i = 1; i < nTime; i++) {
    dp[i] = dp[i - 1];  // 继承上一时刻的最大获利
    auto range = endBy.equal_range(appearedTime[i]);  // 结束时间等于appearedTime[i]的所有的工作 在哈希表中存在的范围
    for_each(range.first, range.second, [&](unordered_multimap<int, int>::value_type& x) {  // 对于在appearedTime[i]结束的每一份工作
        // x.second是这份工作的编号
        // startTime[x.second]是这份工作的开始时间
        // time2loc[startTime[x.second]]是这份工作的开始时间在appearedTime中对应的下标
        // dp[time2loc[startTime[x.second]]是这份工作开始时间的最大获利
        // profit[x.second]是这份工作的获利
        dp[i] = max(dp[i], dp[time2loc[startTime[x.second]]] + profit[x.second]);
    });
}
// dp中的最后一个元素(所有出现过的时刻中的最后一个时刻)即为答案
return dp.back();
  • 时间复杂度O(\nlogn)O(\n log n),其中nn是工作数量,时间复杂度主要来自排序
  • 空间复杂度O(n)O(n),使用了数次哈希表,每次的空间复杂度都是O(n)O(n)

AC代码

C++

class Solution {
public:
    int jobScheduling(vector<int>& startTime, vector<int>& endTime, vector<int>& profit) {
        unordered_set<int> appearedTimeSet;  // 所有的出现过的时间
        unordered_multimap<int, int> endBy;  // <在这个时间结束, 这个任务对应的编号>
        int n = startTime.size();
        for (int i = 0; i < n; i++) {
            appearedTimeSet.insert(startTime[i]);
            appearedTimeSet.insert(endTime[i]);
            endBy.insert({endTime[i], i});
        }
        vector<int> appearedTime;
        for (const int& t : appearedTimeSet) {
            appearedTime.push_back(t);
        }
        sort(appearedTime.begin(), appearedTime.end());
        int nTime = appearedTime.size();
        unordered_map<int, int> time2loc;
        for (int i = 0; i < nTime; i++) {
            time2loc[appearedTime[i]] = i;
        }
        vector<int> dp(nTime);
        for (int i = 1; i < nTime; i++) {
            dp[i] = dp[i - 1];
            auto range = endBy.equal_range(appearedTime[i]);
            for_each(range.first, range.second, [&](unordered_multimap<int, int>::value_type& x) {
                dp[i] = max(dp[i], dp[time2loc[startTime[x.second]]] + profit[x.second]);
            });
            printf("i = %d, appearedTime[%d] = %d, dp[%d] = %d\n", i, i, appearedTime[i], i, dp[i]);  //************
        }
        return dp.back();
    }
};

方法二:二分查找 + DP

方法一中我们使用了数个哈希表将时间和工作映射了起来

方法二学习自力扣官解:leetcode.cn/problems/ma…

这种方法中,dp[i]dp[i]代表前ii份兼职工作可以获得的最大报酬。

因此,dp[i]=max{dp[i1],dp[k]+profit[i1]}dp[i] = \max\{dp[i-1], dp[k] + profit[i - 1]\},其中kk表示结束时间不超过第i1i-1份工作的开始时间的工作数量

这个kk怎么来呢?当然是二分查找

因此我们还需要对工作按照“结束时间”从小到大排个序。

  • 时间复杂度O(\nlogn)O(\n log n),其中nn是工作数量,时间复杂度主要来自排序
  • 空间复杂度O(n)O(n)

AC代码

C++

class Solution {
public:
    int jobScheduling(vector<int>& startTime, vector<int>& endTime, vector<int>& profit) {
        int n = startTime.size();
        vector<vector<int>> jobs(n);
        for (int i = 0; i < n; i++) {
            jobs[i] = {startTime[i], endTime[i], profit[i]};
        }
        sort(jobs.begin(), jobs.end(), [&](const vector<int>& a, const vector<int>& b){
            return a[1] < b[1];
        });
        vector<int> dp(n + 1);
        for (int i = 1; i <= n; i++) {
            int k = upper_bound(jobs.begin(), jobs.begin() + i - 1, jobs[i - 1][0], [&](int st, vector<int>& job) {
                return st < job[1];
            }) - jobs.begin();
            dp[i] = max(dp[i - 1], dp[k] + jobs[i - 1][2]);
        }
        return dp[n];
    }
};

同步发文于CSDN,原创不易,转载请附上原文链接哦~ Tisfy:letmefly.blog.csdn.net/article/det…