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

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

#include<iostream>

#include<cstdio>

#include<cstring>

#include<queue>

#include<algorithm>

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<<u<<" "<<v<<endl;

	add(u,v,w);

	add(v,u,0);

}

bool bfs()

{

	memset(le,0,sizeof(le));

	queue<int>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&&us<f;i=e[i].ne)

		if(le[e[i].to]==le[u]+1&&e[i].v>0)

		{

			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;

}

巴特西

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

最新文章

热门文章