[洛谷201704R1]开心派对小火车

OJ题号：洛谷P3697

思路：

贪心。首先从起点出发，开特急电车，对于每一个特急车站$s_{i}$，分别下一次车，计算从当前车站$s_{i}$出发坐各停电车在指定时限内$t$最远能够到达的车站$r_{i}$，并保证这个$r_{i}$不会超过$s_{i+1}$。将得到的站点$r_{i}$放入一个大根堆中，堆中的每个元素记录两个值，第一个用来保存如果在当前站点开通快速电车，从当前站点坐快速电车总共能到达的站的个数；第二个用来保存当前站的编号$r_i$。最后不断地从堆中挑选最优的站，重复拓展到新的站点，并将新的站点放入堆中。用$ans$保存能够经过的站的个数，注意最后一个拓展的站$r$并不能到达，因此$ans$初值为$-1$。

优化：

用pb_ds的优先队列$(3ms)$比stl的优先队列$(0ms)$快。

#include<cstdio>
#include<queue>
#include<algorithm>
#include<ext/pb_ds/priority_queue.hpp>
const int maxm=3002;
typedef long long ll;
ll n,m,k,a,b,c,t,s[maxm],r[maxm],ans=-1,i;
__gnu_pbds::priority_queue<std::pair<int,int> > pq;
inline void push(const int i) {
    ll v=t-s[i]*b-(r[i]-s[i])*c;
    if(v<0) {
        pq.push(std::make_pair(0,i));
        return;
    }
    ll ri=r[i];
    r[i]=std::min(r[i]+v/a+1,s[i+1]);
    pq.push(std::make_pair(r[i]-ri,i));
}
int main() {
    scanf("%lld%lld%lld%lld%lld%lld%lld",&n,&m,&k,&a,&b,&c,&t);
    for(i=1;i<=m;i++) {
        scanf("%lld",&s[i]);
        s[i]--;
    }
    s[m+1]=n;
    for(i=1;(i<=m)&&(s[i]*b<=t);i++) {
        r[i]=std::min(s[i]+(t-s[i]*b)/a+1,s[i+1]);
        ans+=r[i]-s[i];
        push(i);
    }
    for(i=k-m;i--;) {
        ans+=pq.top().first;
        push(pq.top().second);
        pq.pop();
    }
    printf("%lld",ans);
    return 0;
}