KM模板

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<vector>
using namespace std;  
const int INF=0x3f3f3f3f;  
const int N=2005;  
  
int nx,ny;//两边点数  
int g[N][N];//图 下标从0开始  
int linker[N],lx[N],ly[N];//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(1){  
            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;  
}