描述
后勤部队运来一批武器(机枪和盔甲)。你要把这些武器分配给手下的marine们(每人一部机枪,一套盔甲)。可是问题来了。。。
这些武器的型号不相同(武器是由出价最低的承包商制造的),把一部m型的机枪和一套n型的盔甲分配给一个marine得到的不满意值为(m-n)^2(每个marine当然希望自己得到的武器是同一型号的)。
你的任务就是把a部机枪和b套盔甲分配给手下n个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
第二行: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:费用流 为了满足流量不大于n,多加了一个虚汇点T',T'连向T,最大流量为n
2
Sample 2:
5
//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;
}