堆优化DP 征途堆积出友情的永恒

可以在当前的城市上车,并且在之后的某一座城市下车。从第(i-1)座乘坐到第i座需要花费Ai。在第i座城市上车需要缴纳Bi的税款如果连续地乘坐一段费用大于这次上车前所需缴纳的税款,则这次上车前的税款可以被免除,否则免去乘坐这段火车的费用。
每一次乘坐都不能连续经过超过K座城市(不包括上车时所在的城市),所以,从0~n的最小花费是多少?
【数据范围】
20%的数据满足,N≤30,K≤5。
50%的数据满足,N≤10000,K≤1000。
另有10%的数据满足,Bi = 1。
另有10%的数据满足,Ai = B(i-1)。
另有10%的数据满足,N = K。
100%的数据满足,N≤500000,K≤100000,1≤Ai,Bi≤10^6。

前50% O(N*k)
再30% 特判
100%:堆优化DP
很明显,f[i]=min(f[j]+max(b[j],sum[i]-sum[j]));
对于j,f[j]+b[j]和f[j]-sum[j]都是确定的,
那么维护两个小根堆,维护啥,就在上面。
因为在b[j]>sum[i]-sum[j]时第一个堆里的元素才会合法,所以首先把所有堆1里不合法的pop,但要把它推入堆2.
上面那个操作之前,之后,都要pop在距离上>k的点。

#include <cstdlib>
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
#define N (int)5e5+10
#define inf 100000000000000ll
#define ll long long
using namespace std;
int n,k;ll f[N],a[N],b[N],sum[N];
struct node
{
    int x;ll h;
    node(){}
    node(int x_,ll h_){x=x_;h=h_;}
    friend bool operator <(node x,node y){return x.h>y.h;}
};
priority_queue<node> q1,q2;
int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]),sum[i]=sum[i-1]+a[i];
    for(int i=0;i<n;i++)scanf("%lld",&b[i]);
    memset(f,0x3f,sizeof(f));f[0]=0;
    for(int i=1;i<=n;i++)
    {
        q1.push((node){i-1,f[i-1]+b[i-1]});
        while(!q1.empty())
        {
            node x=q1.top();
            if(x.x>=i-k)break;
            q1.pop();
        }
        while(!q2.empty())
        {
            node x=q2.top();
            if(x.x>=i-k)break;
            q2.pop();
        }
        while(!q1.empty())
        {
            node x=q1.top();
            if(x.h>f[x.x]-sum[x.x]+sum[i])break;
            q1.pop();q2.push((node){x.x,f[x.x]-sum[x.x]});
        }
        while(!q1.empty())
        {
            node x=q1.top();
            if(x.x>=i-k)break;
            q1.pop();
        }
        while(!q2.empty())
        {
            node x=q2.top();
            if(x.x>=i-k)break;
            q2.pop();
        }
        ll tmp=999990000000000000ll;
        if(!q1.empty())tmp=min(tmp,q1.top().h);
        if(!q2.empty())tmp=min(tmp,q2.top().h+sum[i]);
        f[i]=tmp;
    }
    cout<<f[n]<<endl;
}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值