二分图完备匹配下的最大权匹配-KM算法

二分图完备匹配下的最大权匹配-KM算法
题目http://acm.hdu.edu.cn/showproblem.php?pid=2255

#include<stdio.h>
#include<cstring>
using namespace std;
const int N=305;
const int INF=0x3f3f3f3f;
int T;int n;
int nx,ny;      //两边的点数
int g[N][N];    //二分图描述 下标从0开始
int linker[N],lx[N],ly[N];//y中各点匹配状态，x,y中的点标号
int slack[N];
bool visx[N],visy[N];
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(linker[y]==-1||DFS(linker[y])){linker[y]=x;return true;}
        }
        else if(slack[y]>tmp)slack[y]=tmp;
    }
    return false;
}
int KM(){
    memset(linker,-1,sizeof(linker));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(linker[i]!=-1)res+=g[linker[i]][i];
    return res;
}
void init(){
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            scanf("%d",&g[i][j]);
            //若求最小 加上g[i][j]=-g[i][j];
        }
    }
    nx=ny=n;
}
void out(){printf("%d\n",KM());}//若求最小 输出-KM()}
int main()
{
    scanf("%d",&T);//T是测试用例数  hdu2255中T=1
    while(T--){init();out();}
    return 0;
}