【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);
}