P - 奔小康赚大钱 - hdu 2255(带权值的匹配)

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

const int MAXN = 305;
const int oo = 1e9+7;

int w[MAXN][MAXN], N;
int dx[MAXN], dy[MAXN];
int Ly[MAXN], slack[MAXN];
bool vx[MAXN], vy[MAXN];

bool Find(int i)
{///别忘记vx[i]赋值true，表示可以增广的时候达到了i点
    vx[i] = true;
    for(int j=1; j<=N; j++)
    {
        if( !vy[j] && dx[i]+dy[j] == w[i][j] )
        {
            vy[j] = true;

            if( !Ly[j] || Find(Ly[j]) )
            {
                Ly[j] = i;
                return true;
            }
        }
        else if(vy[j] == false)
            slack[j] = min(slack[j], dx[i]+dy[j]-w[i][j]);
    }

    return false;
}
int KM()
{
    int i, j;

    for(i=1; i<=N; i++)
    {
        for(j=1; j<=N; j++)
            slack[j] = oo;

        while(true)
        {
            memset(vx, false, sizeof(vx));
            memset(vy, false, sizeof(vy));

            if( Find(i) == true )break;

            int d = oo;

            for(j=1; j<=N; j++)
            {
                if(!vy[j] && d > slack[j])
                    d = slack[j];
            }

            for(j=1; j<=N; j++)
            {
                if(vx[j])dx[j] -= d;
                if(vy[j])dy[j] += d;
                else slack[j] -= d;
            }
        }
    }

    int sum = 0;

    for(i=1; i<=N; i++)
    {
        sum += w[ Ly[i] ][i];
    }

    return sum;
}

int main()
{
    while(scanf("%d", &N) != EOF)
    {
        memset(dx, false, sizeof(dx));
        memset(dy, false, sizeof(dy));
        memset(Ly, false, sizeof(Ly));

        for(int i=1; i<=N; i++)
        for(int j=1; j<=N; j++)
        {
            scanf("%d", &w[i][j]);
            dx[i] = max(dx[i], w[i][j]);
        }

        printf("%d\n", KM());
    }

    return 0;
}