【JZO5271】【GDOI2018模拟8.14】神奇的救火现场

Description

Data Constraint

Solution

这和以前的jzoj上一道叫保持平衡的题很像啊。当然也有线性的做法，但我们这里只讲一下NlogN的做法。
我们维护两个小根堆分别为水栓和车的供给代价。每遇到一辆车，若水栓不为空，我们从水栓的堆中取出最小的代价x与其坐标相加后加入答案，并将坐标取反-代价加入车堆，若水栓为空，我们则将代价设为+∞重复刚才操作，表示以后若有水栓优先供应给这辆车。每遇到一个水栓，我们都在车里找一个最小答案，若最小答案小于0则加入答案并将坐标取反-代价加入水栓中。否则直接坐标取反加入即可。

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<set>
#define ll long long
using namespace std;
const ll maxn=3e5+5,mo=0x7fffffff;
multiset<ll>f,g;
struct code{
    ll x,bz;
}a[maxn];
ll n,m,i,t,j,k,l,x,y,z,ans;
bool cmp(code x,code y){
    return x.x<y.x;
}
int main(){
    freopen("fire.in","r",stdin);freopen("fire.out","w",stdout);
    scanf("%lld%lld",&m,&n);
    for (i=1;i<=m;i++)scanf("%lld",&a[i].x);
    for (i=1;i<=n;i++)scanf("%lld",&a[m+i].x),a[m+i].bz=1;
    n+=m;
    sort(a+1,a+n+1,cmp);
    for (i=1;i<=n;i++)
        if (a[i].bz){
            if (!f.empty()){
                x=*f.begin();
                t=a[i].x+x;
                ans+=t;
                g.insert(-t-a[i].x);
                f.erase(f.begin());
            }else{
                ans+=mo;
                g.insert(-mo-a[i].x);
            }
        }else{
            if (!g.empty()){
                x=*g.begin();
                if (x+a[i].x<0) g.erase(g.begin());
                else x=-a[i].x;
            }else x=-a[i].x;
            t=a[i].x+x;
            ans+=t;
            f.insert(-t-a[i].x);
        }
    printf("%lld\n",ans);
}