KM算法模板
这个模板是图论书上的那个,书上的代码有错误,在网上各种查找之后finally把这个模板搞对了!match[]数组保存匹配上的左右集合,是对应坐标的,在函数里面开的那个l[]数组是存每个点的具体匹配流量,m,n为左右集合的大小,tu[][]是邻接矩阵,存边的权值。
const int inf=1e9,maxn=510; int KM(int m,int n,int tu[][maxn],int *match1,int *match2) { int s[maxn],t[maxn],l1[maxn],l2[maxn],p,q,ret=0,i,j,k; ///l1为左边的匹配分量,l2是右边的匹配分量 for(i=0; i<m; i++) { for(l1[i]=-inf,j=0; j<n; j++) l1[i]=tu[i][j]>l1[i]?tu[i][j]:l1[i]; if(l1[i]==-inf) return -1; } for(i=0; i<n; l2[i++]=0); memset(match1,-1,sizeof(int)*n); memset(match2,-1,sizeof(int)*n); for(i=0; i<m; i++) { memset(t,-1,sizeof(int)*n); for(s[p=q=0]=i; p<=q&&match1[i]<0; p++) { for(k=s[p],j=0; j<n&&match1[i]<0; j++) if(l1[k]+l2[j]==tu[k][j]&&t[j]<0) { s[++q]=match2[j],t[j]=k; if(s[q]<0) for(p=j; p>=0; j=p) match2[j]=k=t[j],p=match1[k],match1[k]=j; } } if(match1[i]<0) { for(i--,p=inf,k=0; k<=q; k++) for(j=0; j<n; j++) if(t[j]<0&&l1[s[k]]+l2[j]-tu[s[k]][j]<p) p=l1[s[k]]+l2[j]-tu[s[k]][j]; for(j=0; j<n; l2[j]+=t[j]<0?0:p,j++); for(k=0; k<=q; l1[s[k++]]-=p); } } for(i=0; i<m; i++) ret+=tu[i][match1[i]]; return ret; }
持续更新博客地址:
blog.csdn.net/martinue