題目鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=2255

題意：有N個(gè)人跟N個(gè)房子,每個(gè)人跟房子都有一定的距離,現(xiàn)在要讓這N個(gè)人全部回到N個(gè)房子里面去,要求所有人回去的距離最短.

KM算法模板題~

#include "stdio.h" //hdu 2255 #include "string.h" #include "stdlib.h"#define N 305 #define INF 0x3fffffffint n; int match[N]; int map[N][N]; int lx[N],ly[N]; bool s[N],t[N];//s[],t[]記錄當(dāng)前左/右第i個(gè)點(diǎn)是否在匈牙利樹中inline int MIN(int a,int b) { return a<b?a:b; } inline int MAX(int a,int b) { return a>b?a:b; }bool find(int x) //匈牙利,匹配（找增廣路） {int y;s[x] = true;for(y=1; y<=n; ++y){if(!t[y] && lx[x]+ly[y]==map[x][y]){t[y] = true;if(match[y]==-1 || find(match[y])){match[y] = x;return true;}}}return false; }int KM() {int i,j,k;int ans=0;memset(match,-1,sizeof(match));memset(lx,0,sizeof(lx));memset(ly,0,sizeof(ly));for(i=1; i<=n; ++i) //初始化S頂標(biāo)為最大權(quán) {for(j=1; j<=n; ++j)lx[i] = MAX(lx[i],map[i][j]);}for(i=1; i<=n; ++i) //匹配每一個(gè)點(diǎn) {while(1){memset(s,0,sizeof(s));memset(t,0,sizeof(t));if(find(i)) //點(diǎn)i匹配成功 break;else //匹配失敗，找最小值num {int num = INF;for(j=1; j<=n; ++j){if(s[j]) //j在匈牙利樹中 {for(k=1; k<=n; ++k){if(!t[k]) //k在匈牙利樹外num = MIN(num,lx[j]+ly[k]-map[j][k]);}}}for(k=1; k<=n; ++k) //修改頂標(biāo) { if(s[k]) lx[k] -= num; //保證至少有一條邊可以加入if(t[k]) ly[k] += num; //保證原來匹配的邊修改后依然可以匹配 }}}}for(i=1; i<=n; ++i)ans += map[match[i]][i];return ans; }int main() {int i,j;while(scanf("%d",&n)!=-1){for(i=1; i<=n; ++i){for(j=1; j<=n; ++j)scanf("%d",&map[i][j]);}printf("%d\n",KM());}return 0; }

轉(zhuǎn)載于:https://www.cnblogs.com/ruo-yu/p/4411966.html

總結(jié)

以上是生活随笔為你收集整理的hdu 2255 奔小康赚大钱--KM算法模板的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。