完备匹配下的最大权匹配-KM算法的一般模板

#include <iostream>
#include<stdio.h>
#include<cstring>
#include<climits>
#include<algorithm>
using namespace std;
const int maxn=305;
int g[maxn][maxn],lx[maxn],ly[maxn];
int match[maxn];
bool visx[maxn],visy[maxn];
int slack[maxn];
int  n;

bool DFS(int cur)//匈牙利深度搜索算法
{
   visx[cur]=true;
   for(int y=1;y<=n;y++)
   {
    if(visy[y]) continue;
    int t=lx[cur]+ly[y]-g[cur][y];
    if(t==0)
    {
      visy[y]=true;
      if(match[y]==-1||DFS(match[y]))
      {
        match[y]=cur;
        return true;
      }
    }
    else
    slack[y] = min(slack[y],t);  //不在相等子图
   }
return false;
}

int KM_Maxmatch()
{
memset(match,-1,sizeof(match)); //初始化顶标
memset(ly,0,sizeof(ly));//ly[i]为0

for(int i=1;i<=n;i++)
{
lx[i]=-INT_MAX;
for(int j=1;j<=n;j++)//lx[i]为权值最大的边
    {
     if(g[i][j]>lx[i])
     lx[i]=g[i][j];
    }
}
for(int k=1;k<=n;k++) //对n个点匹配
{
for(int m=1;m<=n;m++) slack[m]=INT_MAX;
 while(true)
 {
   memset(visx,false,sizeof(visx));
   memset(visy,false,sizeof(visy));
   if(DFS(k))  break; //匹配成功

  int d=INT_MAX;//匹配失败，找最小值
  for(int i=1;i<=n;i++)//x在交错树中
  {
   if(!visy[i]&&d>slack[i])
    d=slack[i];
  }
  for(int i=1;i<=n;i++)//y在交错树外
  {
   if(visx[i])
    lx[i]-=d;
  }
  for(int i=1;i<=n;i++)//更新顶标
  {
    if(visy[i]) visy[i]+=d;
    else  slack[i]-=d;
  }
 }
}

int result=0; //权值相加
for(int i=1;i<=n;i++)
{
 if(match[i]>-1)
 result+=g[match[i]][i];
}
return result;
}

int main()
{
     int i,j;
     while(scanf("%d",&n)!=EOF)
     {
      for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
           scanf("%d",&g[i][j]);

       int ans=KM_Maxmatch();
       printf("%d\n",ans);
     }
return 0;
}