LUOGU P1315 观光公交(贪心)

Posted 2021-01-17 sdfzsyq

传送门

解题思路

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