【jzoj5043】【保持平衡】【数据结构】

题目大意

博爱路上种起了一棵棵的大树，但是有一些地方的树超过了负荷，有一些地方的树的数量又不够。

我们不妨把博爱路看做一条数轴，数轴有n个点，从1到n编号，第i个位置原来现在有ai棵树，这个位置的需求是bi棵树。ai,bi都是0到10的整数。由于你需要是这个位置的树的数量保持平衡，所以你需要移除或者搬一些树过来。

我们怎么使树的数量平衡呢？

首先，你可以从某个位置i移动一棵树到位置j，这时，你需要的运费是|i-j|*z元。

其次，你可以从商店买一棵树，需要支付x元，这时商店会把树配送到任意位置。

还有就是，你可以叫别人收购在任意位置一棵树，需要支付y元运费。

问使得树的数量平衡最小需要支付多少钱？

解题思路

考虑往右移移的情况，要运走的点，可以花y的费用运走，为了将来可以反悔，我们把-花费-i*z扔进noneed的小根堆。到了需求的点我们可以查询noneed看是否反悔，花费为最小值加i*z。

考虑左移的情况，只需在去最优值的时候相反地考虑即可。

code

#include<set>#include<cmath>#include<cstdio>#include<cstring>#include<algorithm>#define LF double#define LL long long#define min(a,b) ((a<b)?a:b)#define max(a,b) ((a>b)?a:b)#define fo(i,j,k) for(int i=j;i<=k;i++)#define fd(i,j,k) for(int i=j;i>=k;i--)#define fr(i,j) for(int i=begin[j];i;i=next[i])using namespace std;int const Mxn=2*1e5+9,Mxm=16*1e5+9,Inf=1e9;int n,x,y,z;multiset<LL>need,noneed;int main(){    freopen("balance.in","r",stdin);    freopen("balance.out","w",stdout);    scanf("%d%d%d%d",&n,&x,&y,&z);    int ai,bi;LL ans=0,tmp;    fo(i,1,n){        scanf("%d%d",&ai,&bi);        if(ai<bi){            ai=bi-ai;            fo(j,1,ai){                if(!noneed.empty()){                    tmp=(*noneed.begin())+z*i;                    if(tmp<x)noneed.erase(noneed.begin());                    else tmp=x;                }else tmp=x;                ans+=tmp;                need.insert(-i*z-tmp);            }        }        if(ai>bi){            ai-=bi;            fo(j,1,ai){                if(!need.empty()){                    tmp=(*need.begin())+z*i;                    if(tmp<y)need.erase(need.begin());                    else tmp=y;                }else tmp=y;                ans+=tmp;                noneed.insert(-i*z-tmp);            }        }    }    printf("%lld",ans);    return 0;}