把活动按结束时间从小到大排序,对于每个活动求出前面最后一个结束时间<=该活动开始时间的活动,
把活动的结束时间插入到线段树,求前面最后一个<=,就是求前面第一个> 然后-1.
dp[i]=max(dp[i-1],profit[i]+dp[pre[i]]); dp[i]:代表前i个活动安排获得的最大报酬。
class Solution {
public:
struct Segment{
int mx[50000<<2]={0};
void insert(int dex,int val,int l,int r,int pos){
if(l==r) {mx[pos]=val;return;}
int mid=(l+r)>>1;
if(dex<=mid) insert(dex,val,l,mid,pos<<1);
else insert(dex,val,mid+1,r,pos<<1|1);
mx[pos]=max(mx[pos<<1],mx[pos<<1|1]);
}
int query(int val,int l,int r,int pos){
if(l==r) return l;
int mid=(l+r)>>1;
if(mx[pos<<1]>val) return query(val,l,mid,pos<<1);
return query(val,mid+1,r,pos<<1|1);
}
}A;
int jobScheduling(vector<int>& startTime, vector<int>& endTime, vector<int>& profit) {
int n=startTime.size(),*a=new int[n],*pre=new int[n],*dp=new int[n];
for(int i=0;i<n;++i) a[i]=i;
sort(a,a+n,[&](int& x,int &y)->bool{return endTime[x]<endTime[y];});
A.insert(1,endTime[a[0]],1,n,1);
for(int i=1;i<n;++i){
//找到最右边小于等于startTime[a[i]]的
if(A.mx[1]<=startTime[a[i]]) pre[i]=i-1;
else pre[i]=A.query(startTime[a[i]],1,n,1)-2;
A.insert(i+1,endTime[a[i]],1,n,1);//将endTime[a[i]]插入到线段树中
}
dp[0]=profit[a[0]];
for(int i=1;i<n;++i) dp[i]=profit[a[i]]+((pre[i]==-1)?0:dp[pre[i]]),dp[i]=max(dp[i],dp[i-1]);
return dp[n-1];
}
};