LUOGU P1315 观光公交(贪心)

传送门

解题思路

　　首先我们要把加速器一个一个的用，用在什么地方呢。假设当前站车比人晚到，那么车上的人与等待的人都会受到影响；否则的话只有在当前站下车的人有影响。我们第一步先求出车到站的时间，对于一条路来说，如果人等车，那么在这条路用加速器的效果可以继续向后延伸。否则只会对下一站造成影响。这样的话维护一个数组\(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;
}