题目链接:http://soj.me/1303
题目描述:
给出一个完全二分图,并且有|V1|=|V2|,求出最佳匹配。
算法实现:
直接运用二分图最佳匹配算法:Kuhn-Munkras算法。该算法的时间复杂度为O(m*m*n),m、n分别是二分图两个子集的顶点数(m<=n)。对于这道题目而言,m=n并且n<=100,输入规模在可接受范围内。具体实现套用了吉林大学的ACM算法模板。
代码实现:
#include<iostream>
#include<cstring>
#define MAXN 100
#define INF 1e+9
#define clr(x) memset(x,-1,sizeof(int)*MAXN)
using namespace std;
int kuhn_munkras(int m,int n,int mat[][MAXN],int match1[],int match2[])
{
int s[MAXN],t[MAXN],l1[MAXN],l2[MAXN],p,q,ret=0,i,j,k;
for(i=0;i<m;i++)
{
for(l1[i]=-(INF),j=0;j<n;j++)
l1[i]=mat[i][j]>l1[i]?mat[i][j]:l1[i];
if(l1[i]==-(INF)) return -1;
}
for(i=0;i<n;l2[i++]=0);
for(clr(match1),clr(match2),i=0;i<m;i++)
{
for(clr(t),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]==mat[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]-mat[s[k]][j]<p)
p=l1[s[k]]+l2[j]-mat[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++)
{
if(match1[i]<0) return -1;
if(mat[i][match1[i]]<=-(INF)) return -1;
ret+=mat[i][match1[i]];
}
return ret;
}
int main()
{
int n,mat[MAXN][MAXN],match1[MAXN],match2[MAXN];
while(cin>>n&&n>0)
{
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
cin>>mat[i][j];
cout<<kuhn_munkras(n,n,mat,match1,match2)<<endl;
}
return 0;
}