LeetCode第159场周赛：5233. 规划兼职工作（离散化）

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

这里有 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。

示例 3：

输入：startTime = [1,1,1], endTime = [2,3,4], profit = [5,6,4]
输出：6

 

提示：

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

思路：因为开始和结束时间很大，我们可以使用类似于离散化的方式进行映射关系的建立，然后定义两个map，其中一个存储所有时间节点和索引的映射关系，另一个map存结束时间和索引的映射，前一个map注意去重。。。

class Solution {
    public int jobScheduling(int[] startTime, int[] endTime, int[] profit) {
        int n=startTime.length,m=0;
    	Map<Integer,Integer> map=new HashMap<>();
        Map<Integer,List<Integer>> map2=new HashMap<>();
        int[] dp=new int[n*2];
        Set<Integer> set=new HashSet<>();
        for(int i=0;i<n;i++) set.add(startTime[i]);
        for(int i=0;i<n;i++) set.add(endTime[i]);
        Iterator it=set.iterator();
        int[] kp=new int[set.size()];
        while(it.hasNext()) kp[m++]=(int)it.next();
        Arrays.parallelSort(kp);
        for(int i=0;i<m;i++) map.put(kp[i], i+1);
        for(int i=0;i<n;i++){
        	if(map2.containsKey(endTime[i]))
        		map2.get(endTime[i]).add(i);
        	else{
        		map2.put(endTime[i], new ArrayList<>());
        		map2.get(endTime[i]).add(i);
        	}
        }
        for(int i=1;i<=m;i++){
        	dp[i]=dp[i-1];
        	List<Integer> tmp=map2.getOrDefault(kp[i-1], new ArrayList<>());
        	for(int id=0;id<tmp.size();id++)
        		dp[i]=Math.max(dp[i], dp[map.get(startTime[tmp.get(id)])]+profit[tmp.get(id)]);
        }
        return dp[m];
    }
}