tyvj1982 武器分配

描述

    后勤部队运来一批武器(机枪和盔甲)。你要把这些武器分配给手下的marine们(每人一部机枪,一套盔甲)。可是问题来了。。。
    这些武器的型号不相同(武器是由出价最低的承包商制造的),把一部m型的机枪和一套n型的盔甲分配给一个marine得到的不满意值为(m-n)^2(每个marine当然希望自己得到的武器是同一型号的)。
    你的任务就是把a部机枪和b套盔甲分配给手下n个marine。使他们的不满意值之和最小。

输入格式

第一行:3 个正整数 n , a , b (1<=n<=a,b<=80)
第二行:a 个数表示每部机枪的型号
第三行:b 个数表示每套盔甲的型号
0<=型号值<=10000

输出格式

输出一个数:最小不满意值。

测试样例1

输入

Sample 1:
2 3 3
9 10 20
0 10 11
Sample 2:
3 4 4
3 9 7 4
4 2 5 5

输出

Sample 1:
2
Sample 2:
5
费用流 为了满足流量不大于n,多加了一个虚汇点T',T'连向T,最大流量为n
//Serene
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
using namespace std;
const int maxn=300+10,maxm=2*80*80+10,INF=0x3f3f3f3f;
int n,a,b,S,T,we[100],wq[100];

int aa;char cc;
int read() {
	aa=0;cc=getchar();
	while(cc<'0'||cc>'9') cc=getchar();
	while(cc>='0'&&cc<='9') aa=aa*10+cc-'0',cc=getchar();
	return aa;
}

struct Node{
	int x,y,cap,flow,w;
	Node(){}
	Node(int x,int y,int cap,int w) : x(x),y(y),cap(cap),w(w){}
}node[2*maxm];

int fir[maxn],nxt[2*maxm],e=1;
void add(int x,int y,int z,int w) {
	node[++e]=Node(x,y,z,w); nxt[e]=fir[x];fir[x]=e;
	node[++e]=Node(y,x,0,-w); nxt[e]=fir[y];fir[y]=e;
}

int zz[maxn],dis[maxn],from[maxn];
bool vis[maxn];
bool spfa() {
	int s=1,t=0,x,y,z;
	memset(dis,0x3f3f3f3f,sizeof(dis));
	vis[S]=1;zz[++t]=S;dis[S]=0;
	while(s<=t) {
		x=zz[s%maxn];
		for(y=fir[x];y;y=nxt[y]) {
			z=node[y].y;
			if(dis[z]<=dis[x]+node[y].w||node[y].flow>=node[y].cap) continue;
			if(!vis[z]) {
				vis[z]=1; t++;
				zz[t%maxn]=z;
			}
			dis[z]=dis[x]+node[y].w;
			from[z]=y;
		}
		s++; vis[x]=0;
	}
	return dis[T]!=INF;
}

int MCMF() {
	int rs=0,now;
	while(spfa()) {
		now=INF;
		for(int i=T;i!=S;i=node[from[i]].x) now=min(now,node[from[i]].cap-node[from[i]].flow);
		for(int i=T;i!=S;i=node[from[i]].x) {
			node[from[i]].flow+=now;
			node[from[i]^1].flow-=now;
			rs+=node[from[i]].w*now;
		}
	}
	return rs;
}

int main() {
	n=read();a=read();b=read();
	S=a+b+1;T=S+2;
	for(int i=1;i<=a;++i) we[i]=read(),add(S,i,1,0);
	for(int i=1;i<=b;++i) wq[i]=read(),add(a+i,S+1,1,0);
	for(int i=1;i<=a;++i) for(int j=1;j<=b;++j) add(i,a+j,1,(we[i]-wq[j])*(we[i]-wq[j]));
	add(S+1,T,n,0);
	printf("%d",MCMF());
	return 0;
}

  

posted @ 2017-08-28 19:58  shixinyi  阅读(259)  评论(0编辑  收藏  举报