bzoj 3218: a + b Problem【主席树+最小割】

直接建图比较显然,是(s,i,w),(i,t,b),(i,i',p),(i,j,inf),然而建出来之后发现边数是n方级别的,显然跑不过去,然后就有一种比较神的思路:把a离散了建一棵权值线段树,然后要连的j直接放到一个区间内。然而题目又要求j

#include
#include
#include
#include
#include
using namespace std;
const int N=5005,M=500005,inf=1e9;
int n,S,T,ans,h[M],cnt=1,g[N],l[N],r[N],a[N],rt[N],si,tot,le[M];
struct qwe
{
    int ne,to,v;
}e[M];
struct zhuxishu
{
    int l,r;
}t[M];
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void add(int u,int v,int w)
{
    cnt++;
    e[cnt].ne=h[u];
    e[cnt].to=v;
    e[cnt].v=w;
    h[u]=cnt;
}
void ins(int u,int v,int w)
{//cout<q;
    le[S]=1;
    q.push(S);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=h[u];i;i=e[i].ne)
            if(!le[e[i].to]&&e[i].v>0)
            {
                le[e[i].to]=le[u]+1;
                q.push(e[i].to);
            }
    }
    return le[T];
}
int dfs(int u,int f)
{
    if(u==T||!f)
        return f;
    int us=0;
    for(int i=h[u];i&&us0)
        {
            int t=dfs(e[i].to,min(e[i].v,f-us));
            e[i].v-=t;
            e[i^1].v+=t;
            us+=t;
        }
    return us;
}
int dinic()
{
    int re=0;
    while(bfs())
        re+=dfs(S,inf);
    return re;
}
void jia(int x,int ll,int rr,int l,int r,int k)
{
    if(!x)
        return;
    if(ll>=l&&rr<=r)
    {
        ins(k,x,inf);
        return;
    }
    int mid=(ll+rr)>>1;
    if(l<=mid)
        jia(t[x].l,ll,mid,l,r,k);
    if(r>mid)
        jia(t[x].r,mid+1,rr,l,r,k);
}
void update(int u,int x)
{
    int ro=rt[u-1];
    rt[u]=++tot;
    int now=tot,l=1,r=si;
    while(1)
    {
        int mid=(l+r)>>1;
        if(ro)
            ins(now,ro,inf);
        ins(now,u,inf);
        if(l==r)
            break;
        if(x<=mid)
        {
            t[now].l=++tot;
            t[now].r=t[ro].r;
            ro=t[ro].l;
            now=t[now].l;
            r=mid;
        }
        else
        {
            t[now].l=t[ro].l;
            t[now].r=++tot;
            ro=t[ro].r;
            now=t[now].r;
            l=mid+1;
        }
    }
}
int main()
{
    n=read();
    S=0,T=2*n+1;
    for(int i=1;i<=n;i++)
    {
        int b,w,p;
        a[i]=read(),b=read(),w=read(),l[i]=read(),r[i]=read(),p=read();
        g[i]=a[i];
        ans=ans+b+w;
        ins(S,i,b);
        ins(i,T,w);
        ins(i,i+n,p);
    }
    sort(g+1,g+1+n);
    si=unique(g+1,g+1+n)-g-1;
    tot=T;
    for(int i=1;i<=n;i++)
    {
        int le=lower_bound(g+1,g+1+si,l[i])-g,ri=upper_bound(g+1,g+1+si,r[i])-g-1,now=lower_bound(g+1,g+1+si,a[i])-g;
        jia(rt[i-1],1,si,le,ri,i+n);
        update(i,now);
    }
    printf("%d\n",ans-dinic());
    return 0;
}

转载于:https://www.cnblogs.com/lokiii/p/8405407.html

你可能感兴趣的:(bzoj 3218: a + b Problem【主席树+最小割】)