解题思路
首先我们要把加速器一个一个的用,用在什么地方呢。假设当前站车比人晚到,那么车上的人与等待的人都会受到影响;否则的话只有在当前站下车的人有影响。我们第一步先求出车到站的时间,对于一条路来说,如果人等车,那么在这条路用加速器的效果可以继续向后延伸。否则只会对下一站造成影响。这样的话维护一个数组\(rang\),表示影响的范围,然后对人求一个前缀和,每次取一个最大的进行更新。时间复杂度\(O(n*k)\)。
代码
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int MAXN = 100005;
const int inf = 0x3f3f3f3f;
inline int rd(){
int x=0,f=1;char ch=getchar();
while(!isdigit(ch)) {f=ch=='-'?0:1;ch=getchar();}
while(isdigit(ch)) {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
return f?x:-x;
}
int n,m,k,d[MAXN],numw[MAXN],ar[MAXN],lst[MAXN],sum[MAXN];
int ans,rang[MAXN];
struct Data{
int a,b,t;
}data[MAXN];
int main(){
n=rd(),m=rd(),k=rd();
for(int i=1;i<n;i++) d[i]=rd();
for(int i=1;i<=m;i++) {
data[i].t=rd(),data[i].a=rd(),data[i].b=rd();
numw[data[i].b]++;lst[data[i].a]=max(lst[data[i].a],data[i].t);
}
for(int i=1;i<n;i++) ar[i+1]=max(ar[i],lst[i])+d[i];
for(int i=1;i<=n;i++) sum[i]=sum[i-1]+numw[i];lst[n]=inf;
for(int i=1;i<=m;i++)
ans+=ar[data[i].b]-data[i].t;
// for(int i=1;i<=n;i++) cout<<i<<" "<<ar[i]<<endl;
// cout<<ans<<endl;
while(k--){
rang[n-1]=n;
for(int i=n-2;i;i--) {
if(ar[i+1]<=lst[i+1]) rang[i]=i+1;
else rang[i]=rang[i+1];
}
int Max=0,p;
for(int i=1;i<=n;i++)
if(sum[rang[i]]-sum[i]>Max && d[i]){
Max=sum[rang[i]]-sum[i];
p=i;
}
if(!Max) break;
d[p]--;ans-=Max;
ar[0]=0;
for(int i=1;i<n;i++) ar[i+1]=max(ar[i],lst[i])+d[i];
}
printf("%d\n",ans);
return 0;
}