[AGC031]E - Snuke the Phantom Thief 费用流

Solution

假设我们一共选了$k$个，那么假如$x\le a$最多选$b$个，那么$x$第$b+1$大的宝石的$x$必须大于$a$，假如$x\ge a$最多选$b$个，那么$x$第$k-b$大的宝石的$x$必须小于$a$。
也就是说只要枚举选了多少个，那么对于每个宝石的$x、y$都可以得到一个上下界，可以直接跑上下界费用流，但是其实没有必要。因为上下解的范围都很小，所以可以新建一些点表示选择了什么，具体来说，这题可以这样建图：
对于$x,y$分别建$100$个点，对于每个宝石建一个点，向对应的$x,y$连边，其中一条边的费用为宝石的价值，另外一条费用为$0$，两条边流量都为$1$；然后建$2k$个点，分别表示第$i$大的$x$和$y$，然后分别向对应的$x,y$区间连边，都是费用为$0$，流量为$1$，跑最大费用流即可。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int inf=2147483647;
LL read()
{
	LL x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
	return x*f;
}
int n,m,st,ed;
struct P{int x,y;LL v;}p[85];
struct Q{int o,a,b;}q[325];
char op[3];
struct Edge{int x,y,d,next;LL c;}e[100010];
int last[450],len;
void ins(int x,int y,int d,LL c)
{
	int t=++len;
	e[t].x=x;e[t].y=y;e[t].d=d;e[t].c=c;
	e[t].next=last[x];last[x]=t;
}
void addedge(int x,int y,int d,LL c){ins(x,y,d,c),ins(y,x,0,-c);}
LL f[450];int pre[450];bool in[450];
bool spfa()
{
	memset(f,-63,sizeof(f));f[st]=0;
	memset(in,false,sizeof(in));
	queue<int>q;q.push(st);
	while(!q.empty())
	{
		int x=q.front();q.pop();in[x]=false;
		for(int i=last[x];i;i=e[i].next)
		{
			int y=e[i].y;
			if(e[i].d>0&&f[x]+e[i].c>f[y])
			{
				pre[y]=i;
				f[y]=f[x]+e[i].c;
				if(!in[y])in[y]=true,q.push(y);
			}
		}
	}
	if(f[ed]>0)return true;
	return false;
}
LL Ans,ans=0;
void work()
{
	int x=ed;
	while(x!=st)
	{
		Ans+=e[pre[x]].c;
		e[pre[x]].d--,e[pre[x]^1].d++;
		x=e[pre[x]].x;
	}
}
int lx[85],rx[85],ly[85],ry[85];
void solve(int k)
{
	memset(last,0,sizeof(last));len=1;
	for(int i=1;i<=n;i++)
	addedge(p[i].x,200+i,1,p[i].v),addedge(200+i,100+p[i].y,1,0);
	for(int i=1;i<=k;i++)lx[i]=ly[i]=1,rx[i]=ry[i]=100;
	for(int i=1;i<=m;i++)
	{
		if(q[i].o==0)lx[q[i].b+1]=max(lx[q[i].b+1],q[i].a+1);
		if(q[i].o==1&&k>q[i].b)rx[k-q[i].b]=min(rx[k-q[i].b],q[i].a-1);
		if(q[i].o==2)ly[q[i].b+1]=max(ly[q[i].b+1],q[i].a+1);
		if(q[i].o==3&&k>q[i].b)ry[k-q[i].b]=min(ry[k-q[i].b],q[i].a-1);
	}
	for(int i=2;i<=k;i++)lx[i]=max(lx[i],lx[i-1]),ly[i]=max(ly[i],ly[i-1]);
	for(int i=k-1;i;i--)rx[i]=min(rx[i],rx[i+1]),ry[i]=min(ry[i],ry[i+1]);
	st=200+n+2*k+1,ed=st+1;
	for(int i=1;i<=k;i++)
	{
		if(lx[i]>rx[i]||ly[i]>ry[i])return;
		addedge(st,200+n+i,1,0),addedge(200+n+k+i,ed,1,0);
		for(int j=lx[i];j<=rx[i];j++)addedge(200+n+i,j,1,0);
		for(int j=ly[i];j<=ry[i];j++)addedge(100+j,200+n+k+i,1,0);
	}
	int u=0;Ans=0;
	while(spfa())u++,work();
	if(u!=k)return;
	ans=max(ans,Ans);
}
int main()
{
	n=read();
	for(int i=1;i<=n;i++)p[i].x=read(),p[i].y=read(),p[i].v=read();
	m=read();
	for(int i=1;i<=m;i++)
	{
		scanf("%s",op);
		q[i].a=read(),q[i].b=read();
		if(op[0]=='L')q[i].o=0;
		if(op[0]=='R')q[i].o=1;
		if(op[0]=='D')q[i].o=2;
		if(op[0]=='U')q[i].o=3;
	}
	for(int i=1;i<=n;i++)solve(i);
	printf("%lld",ans);
}