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

 

费用流。

有最大流量限制，那么就在汇点T后面加个T2，从T到T2连边，容量为n(最大人数)。

 

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<queue>
#include<cstring>
using namespace std;
const int INF=1e8;
const int mxn=9000;
int read(){
    int 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*10+ch-'0';ch=getchar();}
    return x*f;
}
struct edge{
    int from,v,nxt;
    int f,w;
}e[mxn<<1];
int hd[mxn],mct=1;//
inline void add_edge(int u,int v,int c,int w){
    e[++mct].v=v;e[mct].from=u;e[mct].nxt=hd[u];e[mct].f=c;e[mct].w=w;hd[u]=mct;return;
}
void insert(int u,int v,int c,int w){
    add_edge(u,v,c,w);
    add_edge(v,u,0,-w);
    return;
}
//
int n,m;
int S,T,T2;
int ans=0;
//
int dis[mxn];
bool inq[mxn];
int pre[mxn<<1];
void SPFA(int s){
    memset(dis,0x3f,sizeof dis);
    memset(pre,-1,sizeof pre);
    queue<int>q;
    dis[s]=0;
    inq[s]=1;
    q.push(s);
    while(!q.empty()){
        int u=q.front();q.pop();inq[u]=0;
        for(int i=hd[u];i;i=e[i].nxt){
            if(!e[i].f)continue;
            int v=e[i].v;
            if(dis[v]>dis[u]+e[i].w){
                dis[v]=dis[u]+e[i].w;
                pre[v]=i;//记录前驱边 
                if(!inq[v]){
                    inq[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return;
}
void maxflow(int s,int t){
    SPFA(s);
    while(pre[t]!=-1){
        int tmp=INF;
        for(int i=pre[t];i!=-1;i=pre[e[i].from])
            tmp=min(tmp,e[i].f);
        ans+=dis[t]*tmp;
        for(int i=pre[t];i!=-1;i=pre[e[i].from]){
            e[i].f-=tmp;
            e[i^1].f+=tmp;
        }
        SPFA(s);
    }
    return;
}
int a,b;
int ac[120],bc[120];
int main()
{
    int i,j,u,v;
    n=read();a=read();b=read();
    for(i=1;i<=a;i++){ac[i]=read();}
    for(i=1;i<=b;i++){bc[i]=read();}
    S=0;T=a+b+1;T2=T+1;
    for(i=1;i<=a;i++)
        for(j=1;j<=b;j++){
            insert(i,a+j,1,(ac[i]-bc[j])*(ac[i]-bc[j]));
        }
    for(i=1;i<=a;i++)insert(S,i,1,0);
    for(i=1;i<=b;i++)insert(a+i,T,1,0);
    insert(T,T2,n,0);//限制匹配人数 
    maxflow(S,T2);
    printf("%d\n",ans);
    return 0;
}