bzoj4977 -- 线段树模拟费用流

神题

题解

代码

#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;
}