bzoj4977 -- 線段樹模擬費用流
阿新 • • 發佈:2019-02-04
神題
題解
程式碼
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 100010
struct Node{
int w,f;
}g[N<<1],d[N];
long long Ans;
int i,j,k,n,m,a[N],x,y,h,w[N<<1],c[N<<3],p[N<<3];
inline bool Cmp(Node a,Node b){
if (a.w==b.w)return a.f<b.f;
return a.w<b.w;
}
inline void Down(int x){
if(p[x]){
c[x<<1]+=p[x];p[x<<1]+=p[x];
c[x<<1|1]+=p[x];p[x<<1|1]+=p[x];
p[x]=0;
}
}
inline int Min(int x,int y){
return x<y?x:y;
}
inline void Update(int x,int l,int r,int L,int R,int y){
if(l>R||r<L)return;
if(l>=L&&r<=R){
c[x]+=y;p[x]+=y;
return;
}
Down(x);
int Mid=l+r>>1;
Update(x<<1,l,Mid,L,R,y);
Update(x<<1|1,Mid+1,r,L,R,y);
c[x]=Min(c[x<<1],c[x<<1|1]);
}
inline int Query(int x,int l,int r,int L,int R){
if(l>R||r<L)return 1e9;
if(l>=L&&r<=R)return c[x];
Down(x);
int Mid=l+r>>1;
return Min(Query(x<<1,l,Mid,L,R),Query(x<<1|1,Mid+1,r,L,R));
}
int main(){
scanf("%d%d",&n,&m);
h=n+m;
for(i=1;i<=n;i++)scanf("%d",&a[i]);
sort(a+1,a+n+1);
for(i=1;i<=n;i++)g[i].w=a[i],g[i].f=i;
for(i=1;i<=m;i++)scanf("%d%d",&x,&y),g[i+n].w=x,g[i+n].f=i+n,d[i].w=y-x,d[i].f=i;
sort(g+1,g+h+1,Cmp);sort(d+1,d+m+1,Cmp);
for(i=1;i<=h;i++)w[g[i].f]=i;
for(i=n,j=m;i;i--,j--){
Update(1,1,h,w[i],h,-1);
for(;j;j--){
k=w[d[j].f+n];
if(Query(1,1,h,1,k-1)>=0&&Query(1,1,h,k,h)>=-1&&a[i]+d[j].w>0){
Ans+=a[i]+d[j].w;Update(1,1,h,k,h,1);
break;
}
}
if(!j)break;
}
cout<<Ans<<endl;
return 0;
}