首先对于没有加速器的情况,我们可以O(N)的处理出每一个站的出发时间,然后再O(M)的扫一遍就行。然后再考虑只有一个加速器的情况,我们可以知道,对于i-->i+1的道路使用加速器造成的影响不只是这一段车上的人数,因为可能后面的车站的到达时间因为加速器的使用而提前,使后面站的人等待时间减少,那么我们可以直观的感受到对于i-->i+1的道路使用加速器可以影响到的区间是i到next[i],next[i]表示i后面的车站中第一个需要车等人的车站,那么使用一个加速器对答案产生的贡献就是i-->next[i]中下车的人数,对于查询i到j站有多少人下车,我们可以用前缀和来维护,区间和即为sum[next[i]]-sum[i],表示i+1-->next[i]区间下车人数,也就是在i-->i+1的路径中使用加速器对答案的贡献。由于一共有k个加速器,我们可以做k次上述操作并维护next[i]和到达车站的时间,每次找最大的sun[next[i]]-sum[i],在d[i]处使用,找的过程可以用堆来维护,不过这个数据量暴力枚举也勉强可以。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 10005
int last[maxn],arrive[maxn],d[maxn],sum[maxn];
int fr[maxn],to[maxn],t[maxn],next[maxn],num[maxn];
int n,m,k,ans;
int main()
{
scanf("%d%d%d",&n,&m,&k);
for (int i=1;i<n;i++) scanf("%d",d+i);
for (int i=1;i<=m;i++)
{
scanf("%d%d%d",t+i,fr+i,to+i);
last[fr[i]]=max(last[fr[i]],t[i]);
num[to[i]]++;
}
for (int i=1;i<=n;i++) sum[i]=sum[i-1]+num[i];
for (int i=2;i<=n;i++) arrive[i]=max(arrive[i-1],last[i-1])+d[i-1];
for (int i=1;i<=m;i++) ans=ans+arrive[to[i]]-t[i];
while (k>0)
{
next[n]=n;next[n-1]=n;
for (int i=n-2;i>=1;i--)
if (last[i+1]>=arrive[i+1]) next[i]=i+1;
else next[i]=next[i+1];
int maxi=0,maxd=0;
for (int i=1;i<=n;i++)
if ((sum[next[i]]-sum[i]>maxd)&&(d[i]>0))
{
maxd=sum[next[i]]-sum[i];
maxi=i;
}
if (maxd==0) break;
ans-=maxd;
d[maxi]--;
k--;
for (int i=2;i<=n;i++) arrive[i]=max(arrive[i-1],last[i-1])+d[i-1];
}
printf("%d\n",ans);
return 0;
}