[BZOJ]3218: a + b Problem 主席树优化建图+网络流

Solution

这题算是经典题了……但是一直到今天才做。
本来以为很快就能切了这题，没想到搞了很久，一直有一个点过不去，也不知道问题是什么，反正是有关相同数的合并的问题。
后来不知怎么解决，想到一个不错的方法：离散化，强行让所有数都不同就可以了。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=5010;
const LL inf=(1LL<<61);
const int N=15000;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,st,ed,root[Maxn],lc[Maxn*20],rc[Maxn*20],tot=0;
struct Edge{int y,next;LL d;}e[1500010];
int last[1000010],len=1;
void ins(int x,int y,LL d)
{
    int t=++len;
    e[t].y=y;e[t].d=d;e[t].next=last[x];last[x]=t;
}
void addedge(int x,int y,LL d){/*printf("%d %d %d\n",x,y,d);*/ins(x,y,d),ins(y,x,0);}
int h[1000010];
bool bfs()
{
    memset(h,0,sizeof(h));h[st]=1;
    queue<int>q;q.push(st);
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=last[x];i;i=e[i].next)
        {
            int y=e[i].y;
            if(!h[y]&&e[i].d>0)h[y]=h[x]+1,q.push(y);
        }
    }
    return h[ed];
}
LL dfs(int x,LL f)
{
    if(x==ed)return f;
    LL s=0,t;
    for(int i=last[x];i;i=e[i].next)
    {
        int y=e[i].y;
        if(h[y]==h[x]+1&&e[i].d>0&&s<f)
        {
            t=dfs(y,min(f-s,e[i].d));
            s+=t;e[i^1].d+=t;e[i].d-=t;
        }
    }
    if(s==0)h[x]=0;
    return s;
}
int T;
void insert(int &u,int l,int r,int p,int id)
{
    if(!u)u=++tot;
    if(l==r){addedge(u+T,id,inf);return;}
    int mid=l+r>>1;
    if(p<=mid)insert(lc[u],l,mid,p,id);
    else insert(rc[u],mid+1,r,p,id);
}
int merge(int u1,int u2,int l,int r)
{
    if(!u1||!u2)return u1+u2;
    int mid=l+r>>1;
    lc[u1]=merge(lc[u1],lc[u2],l,mid),rc[u1]=merge(rc[u1],rc[u2],mid+1,r);
    return u1;
}
void link(int u,int l,int r,int fl,int fr,int id)
{
    if(!u)return;
    if(l==fl&&r==fr){addedge(id,u+T,inf);return;}
    int mid=l+r>>1;
    if(fr<=mid)link(lc[u],l,mid,fl,fr,id);
    else if(fl>mid)link(rc[u],mid+1,r,fl,fr,id);
    else link(lc[u],l,mid,fl,mid,id),link(rc[u],mid+1,r,mid+1,fr,id);
}
int a[Maxn],l[Maxn],r[Maxn];
struct Node{int x,id,type;}A[Maxn*3];
bool cmp(Node a,Node b)
{
    if(a.x!=b.x)return a.x<b.x;
    return a.type<b.type;
}
int main()
{
    n=read();T=(n<<1)+2;st=(n<<1)+1,ed=(n<<1)+2;LL ans=0;
    for(int i=1;i<=n;i++)
    {
        int b,w,p;
        A[i*3-2].x=read(),b=read(),w=read(),A[i*3-1].x=read(),A[i*3].x=read(),p=read();
        ans+=(b+w);addedge(st,i,b),addedge(i,ed,w),addedge(i,i+n,p);
    }
    for(int i=1;i<=3*n;i++)
    {
        A[i].id=i;
        if(i%3==2)A[i].type=-1;
        if(i%3==1)A[i].type=0;
        if(i%3==0)A[i].type=1;
    }
    sort(A+1,A+1+3*n,cmp);int cnt=-1;
    for(int i=1;i<=3*n;i++)
    {
        cnt++;
        if(A[i].id%3==1)a[A[i].id/3+1]=cnt;
        if(A[i].id%3==2)l[A[i].id/3+1]=cnt;
        if(A[i].id%3==0)r[A[i].id/3]=cnt;
    }
    root[0]=0;
    for(int i=1;i<=n;i++)insert(root[i],0,N,a[i],i),root[i]=merge(root[i],root[i-1],0,N);
    for(int i=1;i<=tot;i++)
    {
        if(lc[i])addedge(i+T,lc[i]+T,inf);
        if(rc[i])addedge(i+T,rc[i]+T,inf);
    }
    for(int i=2;i<=n;i++)link(root[i-1],0,N,l[i],r[i],i+n);
    while(bfs())ans-=dfs(st,inf);
    printf("%lld",ans);
}