【bzoj3218】 a + b Problem 最小割+主席树

好题!!!

最小割
正常做法
与S联通表示白色,与T联通表示黑色
每个点i拆成两个点,i和i'
源点S向i连一条容量为wi的边
i向汇点T连一条容量为bi的边
i'向i连一条容量为p的边
所有满足条件的j向i'连一条容量为inf的边
(因为条件是或,所以要这样建图)
但是边数太多了!!!!
线段树优化连边?(自己YY的)
对原序列建主席树,我们发现每个点的区间其实对应着主席树上的O(logn)个节点
不妨建出一棵主席树,每个点向新建出的一条链连一条容量为inf的边
同时,上次的节点向现在的节点连一条容量为inf的边
每次每个节点查询到的O(logn)个主席树上的节点对应连边


点数 2n+nlogn 200000
边数 (4n+2nlogn)*2 1000000


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#define maxn 200010
#define maxm 1000010
#define inf 1000000000

using namespace std;

struct yts
{
	int x,id;
}e[5010];

int head[maxn],to[maxm],c[maxm],next[maxm],q[maxn],d[maxn];
int lch[maxn],rch[maxn],root[maxn],cnt[maxn];
int L[5010],R[5010],key[5010],seq[5010];
int n,m,num,tot,s,t,ans,mx;

void addedge(int x,int y,int z)
{
	num++;to[num]=y;c[num]=z;next[num]=head[x];head[x]=num;
	num++;to[num]=x;c[num]=0;next[num]=head[y];head[y]=num;
}

int findl(int x)
{
	int l=1,r=mx,ans=mx;
	while (l<=r)
	{
		int mid=(l+r)/2;
		if (seq[mid]>=x) ans=mid,r=mid-1; else l=mid+1;
	}
	return ans;
}

int findr(int x)
{
	int l=1,r=mx,ans=1;
	while (l<=r)
	{
		int mid=(l+r)/2;
		if (seq[mid]<=x) ans=mid,l=mid+1; else r=mid-1;
	}
	return ans;
}

int modify(int pre,int l,int r,int i)
{
	int now=++tot;
	if (l==r)
	{
		lch[now]=rch[now]=0;cnt[now]=cnt[pre]+1;
	}
	else
	{
		int mid=(l+r)/2;
		if (key[i]<=mid)
		{
			lch[now]=modify(lch[pre],l,mid,i);rch[now]=rch[pre];
		}
		else 
		{
			rch[now]=modify(rch[pre],mid+1,r,i);lch[now]=lch[pre];
		}
		cnt[now]=cnt[lch[now]]+cnt[rch[now]];
	}
	addedge(i,2*n+1+now,inf);
	addedge(2*n+1+pre,2*n+1+now,inf);
	return now;
}

void query(int root,int l,int r,int i)
{
	if (!cnt[root]) return;
	if (L[i]<=l && r<=R[i])
	{
		addedge(2*n+1+root,n+i,inf);
		return;
	}
	int mid=(l+r)/2;
	if (L[i]<=mid) query(lch[root],l,mid,i);
	if (mid<R[i]) query(rch[root],mid+1,r,i);
}

bool bfs()
{
	memset(d,-1,sizeof(d));
	int l=0,r=1;
	q[1]=s;d[s]=0;
	while (l<r)
	{
		int x=q[++l];
		for (int p=head[x];p;p=next[p])
		  if (c[p] && d[to[p]]==-1)
		  {
		  	d[to[p]]=d[x]+1;
		  	q[++r]=to[p];
		  }
	}
	if (d[t]==-1) return 0; else return 1;
}

int find(int x,int low)
{
	if (x==t || low==0) return low;
	int totflow=0;
	for (int p=head[x];p;p=next[p])
	  if (c[p] && d[to[p]]==d[x]+1)
	  {
	  	int a=find(to[p],min(low,c[p]));
	  	c[p]-=a;c[p^1]+=a;
	  	low-=a;totflow+=a;
	  	if (low==0) return totflow;
	  }
	if (low) d[x]=-1;
	return totflow;
}

bool cmp(yts x,yts y)
{
	return x.x<y.x;
}

int main()
{
	scanf("%d",&n);
	num=1;s=0;t=100000;
	for (int i=1;i<=n;i++)
	{
		int b,w,p;
		scanf("%d%d%d%d%d%d",&e[i].x,&b,&w,&L[i],&R[i],&p);
		e[i].id=i;ans+=b+w;
		addedge(s,i,w);
		addedge(i,t,b);
		addedge(n+i,i,p);
	}
	sort(e+1,e+n+1,cmp);
	mx=1;key[e[1].id]=1;seq[1]=e[1].x;
	for (int i=2;i<=n;i++)
	  if (e[i].x==e[i-1].x) key[e[i].id]=mx;
	  else key[e[i].id]=++mx,seq[mx]=e[i].x;
	tot=0;root[0]=lch[0]=rch[0]=0;
	for (int i=1;i<=n;i++)
	{
		L[i]=findl(L[i]);R[i]=findr(R[i]);
		root[i]=modify(root[i-1],1,mx,i);
		query(root[i-1],1,mx,i);		
	}
	while (bfs()) ans-=find(s,inf);
	printf("%d\n",ans);
	return 0;
}


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