2019 ICPC Asia Nanjing Regional J Spy 二分图带权最大匹配 KM

题目链接

#include <bits/stdc++.h>

const int maxn = 400 + 5;
const int inf=0x3f3f3f3f;

long long a[maxn],p[maxn],b[maxn],c[maxn],n;
struct KM
{
	#define type long long
	static const int N=405;
	int n,mx[N],my[N],prv[N];
	type slk[N],lx[N],ly[N],w[N][N];
	bool vx[N],vy[N];
	void init(int _n)
	{
		n=_n;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				w[i][j]=0;
			}
		}
	}
	void add_edge(int x,int y,type val){w[x][y]=val;}
	void match(int y){while(y) std::swap(y,mx[my[y]=prv[y]]);}
	void bfs(int x)
	{
		int i,y;
		type d;
		for(i=1;i<=n;i++)
		{
			vx[i]=vy[i]=0;
			slk[i]=inf;
		}
		std::queue<int> q;
		q.push(x);
		vx[x]=1;
		while(1)
		{
			while(!q.empty())
			{
				x=q.front();
				q.pop();
				for(y=1;y<=n;y++)
				{
					d=lx[x]+ly[y]-w[x][y];
					if(!vy[y]&&d<=slk[y])
					{
						prv[y]=x;
						if(!d)
						{
							if(!my[y]) return match(y);
							q.push(my[y]);
							vx[my[y]]=1;
							vy[y]=1;
						}
						else slk[y]=d;
					}
				}
			}
			d=inf+1;
			for(i=1;i<=n;i++)
			{
				if(!vy[i]&&slk[i]<d)
				{
					d=slk[i];
					y=i;
				}
			}
			for(i=1;i<=n;i++)
			{
				if(vx[i]) lx[i]-=d;
				if(vy[i]) ly[i]+=d;
				else slk[i]-=d;
			}
			if(!my[y]) return match(y);
			q.push(my[y]);
			vx[my[y]]=1;
			vy[y]=1;
		}
	}
	type max_match()
	{
		int i;
		type res;
		for(i=1;i<=n;i++)
		{
			mx[i]=my[i]=ly[i]=0;
			lx[i]=*std::max_element(w[i]+1,w[i]+n+1);
		}
		for(i=1;i<=n;i++) bfs(i);
		res=0;
		for(i=1;i<=n;i++) res+=lx[i]+ly[i];
		return res;
	}
	#undef type
	#undef inf
}km;

int main()
{
#ifdef LOCAL
	freopen("input.in","r",stdin);
#endif
	scanf("%lld",&n); km.init(n);
	for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
	for(int i=1;i<=n;i++)scanf("%lld",&p[i]);
	for(int i=1;i<=n;i++)scanf("%lld",&b[i]);
	for(int i=1;i<=n;i++)scanf("%lld",&c[i]);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			int sum=0;
			for(int k=1;k<=n;k++)
				if(b[i]+c[j]>a[k]) sum+=p[k];
			km.add_edge(i,j,sum);
		}
	printf("%lld\n",km.max_match());
	return 0;
}