KM算法模板

来源:互联网 发布:华为荣耀4a手机壳淘宝 编辑:程序博客网 时间:2024/05/19 19:59


这个模板是图论书上的那个,书上的代码有错误,在网上各种查找之后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;}


0 0
原创粉丝点击