TYVJ1982 武器分配

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 #include<cmath>
 5 #include<queue>
 6 #include<cstring>
 7 using namespace std;
 8 const int INF=1e8;
 9 const int mxn=9000;
10 int read(){
11     int x=0,f=1;char ch=getchar();
12     while(ch<‘0‘ || ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
13     while(ch>=‘0‘ && ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();}
14     return x*f;
15 }
16 struct edge{
17     int from,v,nxt;
18     int f,w;
19 }e[mxn<<1];
20 int hd[mxn],mct=1;//
21 inline void add_edge(int u,int v,int c,int w){
22     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;
23 }
24 void insert(int u,int v,int c,int w){
25     add_edge(u,v,c,w);
26     add_edge(v,u,0,-w);
27     return;
28 }
29 //
30 int n,m;
31 int S,T,T2;
32 int ans=0;
33 //
34 int dis[mxn];
35 bool inq[mxn];
36 int pre[mxn<<1];
37 void SPFA(int s){
38     memset(dis,0x3f,sizeof dis);
39     memset(pre,-1,sizeof pre);
40     queue<int>q;
41     dis[s]=0;
42     inq[s]=1;
43     q.push(s);
44     while(!q.empty()){
45         int u=q.front();q.pop();inq[u]=0;
46         for(int i=hd[u];i;i=e[i].nxt){
47             if(!e[i].f)continue;
48             int v=e[i].v;
49             if(dis[v]>dis[u]+e[i].w){
50                 dis[v]=dis[u]+e[i].w;
51                 pre[v]=i;//记录前驱边 
52                 if(!inq[v]){
53                     inq[v]=1;
54                     q.push(v);
55                 }
56             }
57         }
58     }
59     return;
60 }
61 void maxflow(int s,int t){
62     SPFA(s);
63     while(pre[t]!=-1){
64         int tmp=INF;
65         for(int i=pre[t];i!=-1;i=pre[e[i].from])
66             tmp=min(tmp,e[i].f);
67         ans+=dis[t]*tmp;
68         for(int i=pre[t];i!=-1;i=pre[e[i].from]){
69             e[i].f-=tmp;
70             e[i^1].f+=tmp;
71         }
72         SPFA(s);
73     }
74     return;
75 }
76 int a,b;
77 int ac[120],bc[120];
78 int main()
79 {
80     int i,j,u,v;
81     n=read();a=read();b=read();
82     for(i=1;i<=a;i++){ac[i]=read();}
83     for(i=1;i<=b;i++){bc[i]=read();}
84     S=0;T=a+b+1;T2=T+1;
85     for(i=1;i<=a;i++)
86         for(j=1;j<=b;j++){
87             insert(i,a+j,1,(ac[i]-bc[j])*(ac[i]-bc[j]));
88         }
89     for(i=1;i<=a;i++)insert(S,i,1,0);
90     for(i=1;i<=b;i++)insert(a+i,T,1,0);
91     insert(T,T2,n,0);//限制匹配人数 
92     maxflow(S,T2);
93     printf("%d\n",ans);
94     return 0;
95 }