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

本来以为是强行加主席树

没想到是主席树优化建边 太神了!!


注意主席树叶子节点的特判 wa了1h+!

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm> 
using namespace std;
const int MAXN=5000+10;
const int MAXM=5000*100;
const int INF=1000000000;
int tot=1,g[MAXM],nnext[MAXM],flow[MAXM],num[MAXM];
void add(int x,int y,int z){tot++;nnext[tot]=g[x];g[x]=tot;num[tot]=y;flow[tot]=z;}
void ADD(int x,int y,int z){add(x,y,z);add(y,x,0);}
struct H{int a,b,w,l,r,p;}c[MAXN];
int n,s,t,cnt;
int team[MAXM],head,tail,d[MAXM];
int ls[MAXN*3],id[MAXN*3]; 
bool bfs()
{
	memset(d,0,sizeof(d));
	head=tail=0;d[s]=1;
	team[++tail]=s;
	while(head<tail)
	{
		int x=team[++head];
		for(int i=g[x];i;i=nnext[i])
			if(flow[i]>0&&d[num[i]]==0)
			{
				d[num[i]]=d[x]+1;
				team[++tail]=num[i];
			}
	}
	if(d[t]==0) return false;
	return true;
}
int dfs(int x,int mmin)
{
	if(x==t) return mmin;
	int f=0,tmp;
	for(int i=g[x];i;i=nnext[i])
		if(d[num[i]]==d[x]+1&&flow[i]>0&&(tmp=dfs(num[i],min(mmin,flow[i]))))
		{
			flow[i]-=tmp;
			flow[i^1]+=tmp;
			f+=tmp;mmin-=tmp;
			if(mmin==0) return f;
		}
	if(f==0) d[x]=0;
	return f;
}
int find(int x)
{
	int l=1,r=3*n;
	while(l<r)
	{
		int mid=(l+r)/2;
		if(ls[mid]>=x) r=mid;
		else l=mid+1;
	}
	return id[l];
}
int root[MAXN],opt;
struct HH {int l,r,num;}seg[MAXM];
void build(int &now,int last,int l,int r,int x)
{	
	now=++opt;
	if(l==r) 
	{
		seg[now].num=++cnt;
		ADD(seg[now].num,seg[last].num,INF); 
		ADD(seg[now].num,x,INF);
		return ;
	}
	seg[now].num=++cnt;
	int mid=(l+r)/2;
	if(c[x].a<=mid) 
	{
		seg[now].r=seg[last].r; 
		build(seg[now].l,seg[last].l,l,mid,x);
	}
	else
	{
		seg[now].l=seg[last].l; 
		build(seg[now].r,seg[last].r,mid+1,r,x);
	}
	if(seg[now].l!=0) ADD(seg[now].num,seg[seg[now].l].num,INF);
	if(seg[now].r!=0) ADD(seg[now].num,seg[seg[now].r].num,INF);
}
void link(int now,int l,int r,int s,int t,int i)
{
	if(now==0) return ;
	if(s<=l&&r<=t)
	{
		ADD(i,seg[now].num,INF);
		return ;
	}
	int mid=(l+r)/2;
	if(s<=mid) link(seg[now].l,l,mid,s,t,i);
	if(mid+1<=t) link(seg[now].r,mid+1,r,s,t,i);
}
int main()
{
	scanf("%d",&n);cnt=2*n;s=++cnt,t=++cnt;int ans=0;
	for(int i=1;i<=n;i++) scanf("%d %d %d %d %d %d",&c[i].a,&c[i].b,&c[i].w,&c[i].l,&c[i].r,&c[i].p),ans+=c[i].w+c[i].b,ls[i]=c[i].l,ls[i+n]=c[i].r,ls[i+n*2]=c[i].a;
	for(int i=1;i<=n;i++) ADD(i,i+n,c[i].p),ADD(s,i,c[i].b),ADD(i,t,c[i].w);
	sort(ls+1,ls+3*n+1);	
	id[1]=1;for(int i=2;i<=3*n;i++) if(ls[i]!=ls[i-1]) id[i]=id[i-1]+1; else id[i]=id[i-1];
	for(int i=1;i<=n;i++) c[i].l=find(c[i].l),c[i].r=find(c[i].r),c[i].a=find(c[i].a);
	for(int i=1;i<=n;i++) link(root[i-1],1,id[3*n],c[i].l,c[i].r,i+n),build(root[i],root[i-1],1,id[3*n],i);
	while(bfs()) ans-=dfs(s,INF);
	printf("%d\n",ans);
	return 0;
}

    原文作者:B树
    原文地址: https://blog.csdn.net/sxb_201/article/details/51254138
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞