【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
#include
#include
#include
#include
#include
#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