KM算法模板

View Code

#define MIN(a,b) ((a)<(b)?(a):(b))
#define INF 999999
#define MAX 5110
int n,match[MAX];
bool sx[MAX],sy[MAX];
int lx[MAX],ly[MAX],map[MAX][MAX];
bool path(int u)
{
    sx[u]=true;
    for(int v=0;v<n;v++)
        if(!sy[v]&&lx[u]+ly[v]==map[u][v])
        {
            sy[v]=true;
            if(match[v]==-1||path(match[v]))
            {
                match[v]=u;
                return true;
            }
        }
        return false;
}
int KM(bool truth)//可以不用更改地处理最小或最大权匹配
{
    int i,j;
    if(!truth)
    {
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
                map[i][j]=-map[i][j];
    }
    for(i=0;i<n;i++)
    {
        lx[i]=-INF;
        ly[i]=0;
        for(j=0;j<n;j++)
            if(lx[i]<map[i][j])
                lx[i]=map[i][j];
    }
    memset(match,-1,sizeof(match));
    for(int u=0;u<n;u++)
        while(1)
        {
            memset(sx,0,sizeof(sx));
            memset(sy,0,sizeof(sy));
            if(path(u)) break;
            int dmin=INF;
            for(i=0;i<n;i++)
                if(sx[i])
                    for(j=0;j<n;j++)
                        if(!sy[j])
                            dmin=MIN(lx[i]+ly[j]-map[i][j],dmin);
                        for(i=0;i<n;i++)
                        {
                            if(sx[i])
                                lx[i]-=dmin;
                            if(sy[i])
                                ly[i]+=dmin;
                        }
        }
        int sum=0;
        for(j=0;j<n;j++)
            sum+=map[match[j]][j];
        if(!truth)
        {
            sum=-sum;
            for(i=0;i<n;i++)
                for(j=0;j<n;j++)
                    map[i][j]=-map[i][j];
        }
        return sum;
}
void Map_Init(int m,int n)
{
    int x,y,w;
    int i,j;
    for(i=0;i<n;i++)
        for(j=0;j<n;j++)
            map[i][j]=INF;
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&x,&y,&w);
            if(map[x-1][y-1]>w)
                map[x-1][y-1]=w;
        }
}

编辑器加载中...