描述
后勤部队运来一批武器(机枪和盔甲)。你要把这些武器分配给手下的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; }