HDU 2255 KM 최적화 (n ^ 3) 템 플 릿

다음으로 이동:http://blog.csdn.net/cnyali/article/details/50663663

#include <iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
const int maxn = 101;
const int INF = (1<<31)-1;
int w[maxn][maxn];
int lx[maxn],ly[maxn]; //  
int match[maxn];
int visx[maxn],visy[maxn];
int slack[maxn];
int nx,ny;
bool find(int x)
{
    visx[x] = true;
    for(int y = 0; y < ny; y++)
    {
        if(visy[y])
            continue;
        int t = lx[x] + ly[y] - w[x][y];
        if(t==0)
        {
            visy[y] = true;
            if(match[y]==-1 || find(match[y]))
            {
                match[y] = x;
                return true;        //     
            }
        }
        else if(slack[y] > t)
            slack[y] = t;
    }
    return false;                   //       （    x       ，     (         )  ）
}

int KM()                //        
{
    int i,j;

    memset(match,-1,sizeof(match));
    memset(ly,0,sizeof(ly));
    for(i = 0; i < nx; i++)
        for(j = 0,lx[i] = -INF; j < ny; j++)
            if(w[i][j] > lx[i])
                lx[i] = w[i][j];
    for(int x = 0; x < nx; x++)
    {
        for(i = 0; i < ny; i++)
            slack[i] = INF;
        while(true)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(find(x))                     //     ，  
                break;
            int d = INF;
            for(i = 0; i < ny; i++)          //   ， l   (          )，   
            {
                if(!visy[i] && d > slack[i])
                    d = slack[i];
            }
            for(i = 0; i < nx; i++)
            {
                if(visx[i])
                    lx[i] -= d;
            }
            for(i = 0; i < ny; i++)
            {
                if(visy[i])
                     ly[i] += d;
                else
                     slack[i] -= d;
            }
        }
    }
    int result = 0;
    for(i = 0; i < ny; i++)
    if(match[i]>-1)
        result += w[match[i]][i];
    return result;
}

int main()
{
   // freopen("g:/1.txt","r",stdin);
    while(true)
    {
        scanf("%d%d",&nx,&ny);
        int a,b,c;
        while(scanf("%d%d%d",&a,&b,&c),a+b+c)
        {
            w[a][b]=c;
        }
        printf("%d
",KM());
        break;
    }
    return 0;
}