KM算法
2016-01-31 09:07
393 查看
#include <stdio.h> #include <algorithm> #include <string.h> #include <iostream> using namespace std; /* KM算法 * 复杂度O(nx*nx*ny) * 求最大权匹配 * 若求最小权匹配,可将权值取相反数,结果取相反数 * 点的编号从0开始 */ const int N = 310; const int INF = 0x3f3f3f3f; int nx,ny;//两边的点数 int g ;//二分图描述 int match ,lx ,ly ;//y中各点匹配状态,x,y中的点标号 int slack ; bool visx ,visy ; bool DFS(int x) { visx[x] = true; for(int y = 0; y < ny; y++) { if(visy[y])continue; int tmp = lx[x] + ly[y] - g[x][y]; if(tmp == 0) { visy[y] = true; if(match[y] == -1 || DFS(match[y])) { match[y] = x; return true; } } else if(slack[y] > tmp) slack[y] = tmp; } return false; } int KM() { memset(match,-1,sizeof(match)); memset(ly,0,sizeof(ly)); for(int i = 0;i < nx;i++) { lx[i] = -INF; for(int j = 0;j < ny;j++) if(g[i][j] > lx[i]) lx[i] = g[i][j]; } for(int x = 0;x < nx;x++) { for(int i = 0;i < ny;i++) slack[i] = INF; while(true) { memset(visx,false,sizeof(visx)); memset(visy,false,sizeof(visy)); if(DFS(x))break; int d = INF; for(int i = 0;i < ny;i++) if(!visy[i] && d > slack[i]) d = slack[i]; for(int i = 0;i < nx;i++) if(visx[i]) lx[i] -= d; for(int i = 0;i < ny;i++) { if(visy[i])ly[i] += d; else slack[i] -= d; } } } int res = 0; for(int i = 0;i < ny;i++) if(match[i] != -1) res += g[match[i]][i]; return res; }
相关文章推荐
- 自定义旋转的ProgressBar
- Cocos2d-x 3.1.1 Lua实例-AccelerometerTest(重力加速计)
- 苹果 企业 开发者账号 注册、申请续费整个流程($299的账号)
- android项目中刷新activity界面
- UESTC-888-Absurdistan Roads(kruskal+floyd)
- [LeetCode] Binary Tree Upside Down 二叉树的上下颠倒
- Java设计模式编程中的工厂方法模式和抽象工厂模式
- 数学符号及读法大全
- HDOJ 5616-Jam's balance【模拟】
- Eclipse下通过Maven的Jetty插件运行Web工程的配置
- 把字符串转换为整数
- 通过条件的查询----涉及到三层(Action Service DAO)
- 2015杂谈
- poj 1704 Georgia and Bob(阶梯博弈)
- 奔小康赚大钱 hdu 2255
- ios-提升之【1】-UIApplication
- 2016太原网络营销师郭文军分享网站中如何合理布局关键词?
- vs2012 Nuget错误:“未能解析此远程名称api.nuget.org”
- RadioGroup方法总结
- caffe 卷积层的运算