hdu 1533 Going Home (二分图最佳匹配,KM算法)
题意:给出一个n*m的maze,其中包含不超过一个人(用m表示),以及和人数相等的房子(用H表示),其他都是‘.’,表示可以经过的路径。人向一个方向移动花费代价1.问每个人都回到一个房子里的最小代价是多少。ps:每个格子是无限大的,也就是所有人可以同时踩在一个格子里。以及:路过一个房子可以不住,而只是“经过”。
思路:有了那两个条件,这题就是赤裸的二分图最优匹配了。建图也很easy.可以预处理下w,就是两点的哈密顿距离.
需要注意的是这道题求的是最小权值。那么做法就是将w取负,然后答案再次取负即可。
(还有其他方法处理,不过这种最easy应该?)
(不过要保证初始化某些数组的时候要比所有的值小,所以不能是0,而应该是-inf)
以及:初始化数组为-inf的方法,可以memset(lx,0xc0,sizeof(lx));
这样得到的-inf只和inf的绝对值差1,好评如潮。
0xc0,0xc0,0xc0,重要的常数说三遍
哦还有,不要忘记在每次find的时候记录X集合中的点,这是比hungary算法的find里多的一个步骤,并且是容易忘记的。。。
1/* ***********************************************
2Author :111qqz
3Created Time :2016年06月01日 星期三 16时45分41秒
4File Name :code/hdu/1533.cpp
5************************************************ */
6
7#include <cstdio>
8#include <cstring>
9#include <iostream>
10#include <algorithm>
11#include <vector>
12#include <queue>
13#include <set>
14#include <map>
15#include <string>
16#include <cmath>
17#include <cstdlib>
18#include <ctime>
19#define fst first
20#define sec second
21#define lson l,m,rt<<1
22#define rson m+1,r,rt<<1|1
23#define ms(a,x) memset(a,x,sizeof(a))
24typedef long long LL;
25#define pi pair < int ,int >
26#define MP make_pair
27
28using namespace std;
29const double eps = 1E-8;
30const int dx4[4]={1,0,0,-1};
31const int dy4[4]={0,-1,1,0};
32const int inf = 0x3f3f3f3f;
33const int N=105;
34int n,m;
35char maze[N][N];
36int tot1,tot2;
37int w[N][N];
38int link[N];
39bool visx[N],visy[N];
40int lx[N],ly[N];
41int slk[N];
42struct point
43{
44 int x,y;
1 point(){}
2 point(int _x,int _y):
3 x(_x),y(_y){};
1 int dis(point b)
2 {
3 int res;
4 res = abs(x-b.x)+abs(y-b.y);
5 return -res;
6 }
7}p1[N*N],p2[N*N];
8
9void init()
10{
11 tot1 = tot2 = 0 ;
12 ms(w,0xc0); //初始化为负无穷。。。0xc0这个数吼啊。。。和inf的0x3f的绝对值就差1.
13 // cout<<"-inf:"<<w[1][1]<<endl;
14 // cout<<"inf:"<<inf<<endl;
15}
16
17bool find(int u)
18{
19
20 visx[u] = true; //不要忘记标记X集合中找过增广路的点。
21 for ( int v = 1 ; v <= tot2 ; v++)
22 {
23 if (visy[v]) continue;
24 int tmp = lx[u] + ly[v] - w[u][v];
25 if (tmp==0)
26 {
27 visy[v] = true;
28
29 if (link[v]==-1||find(link[v]))
30 {
31 link[v] = u;
32 return true;
33 }
34 }
35 else
36 if (tmp<slk[v]) slk[v] = tmp ;
37 }
38 return false;
39}
40
41int KM()
42{
43 ms(lx,0xc0); //因为所有的权值都是负的(?,所以初始化不能为0而是负无穷。
44 ms(ly,0);
45 ms(link,-1);
46
47 for ( int i = 1 ; i <= tot1 ; i ++)
48 for ( int j = 1 ; j <= tot2 ; j++)
49 lx[i] = max(w[i][j],lx[i]);
50
51 // cout<<"tot1:"<<tot1<<" tot2:"<<tot2<<endl;
52
53 // for ( int i = 1 ; i <= tot1 ; i ++) cout<<"lx[i]:"<<lx[i]<<" ly[i]:"<<ly[i]<<endl;
54 for ( int i = 1 ; i <= tot1 ; i++)
55 {
56 ms(slk,0x3f);
57
58 while (1)
59 {
60 ms(visx,false);
61 ms(visy,false);
62
63 if (find(i)) break;
64
65 int d = inf;
66 for ( int j = 1 ; j <= tot1 ; j++)
67 {
68 if (!visy[j]&&slk[j]<d) d = slk[j];
69 }
70
71 for ( int j = 1 ; j <= tot1 ; j++) if (visx[j]) lx[j]-=d;
72 for ( int j = 1 ; j <= tot2 ; j++) if (visy[j]) ly[j]+=d;else slk[j]-=d;
73 }
74 }
75
76
77 int res = 0 ;
78
79 for ( int i = 1 ; i <= tot1 ; i ++)
80 if (link[i]>-1) res += w[link[i]][i];
81
82 return -res;
83
84
85}
86int main()
87{
88 #ifndef ONLINE_JUDGE
89 freopen("code/in.txt","r",stdin);
90 #endif
91 while (scanf("%d%d",&n,&m)!=EOF)
92 {
93 if (n==0&&m==0) break;
94
95 init();
96 for ( int i = 0 ; i < n ; i++) scanf("%s",maze[i]);
97
98 for ( int i = 0 ; i < n ; i++)
99 for ( int j = 0 ; j < m ; j++)
100 if (maze[i][j]=='m') p1[++tot1] = point(i,j);
101 else if (maze[i][j]=='H')p2[++tot2] = point(i,j);
102
103 for ( int i = 1 ; i <= tot1 ; i ++)
104 for ( int j = 1; j <= tot2 ; j++)
105 w[i][j] = p1[i].dis(p2[j]);
106
107 // cout<<"wwwwwwwwwwwwW?????"<<endl;
108 int ans = KM();
109 // cout<<"kkkkkkkkkkkkkkkkk"<<endl;
110 printf("%d\n",ans);
111
112 }
113
114 #ifndef ONLINE_JUDGE
115 fclose(stdin);
116 #endif
117 return 0;
118}