1. 程式人生 > >bzoj4977 -- 線段樹模擬費用流

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