HDU 2255 奔小康赚大钱(KM算法求完备匹配)

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define INF 0x3f3f3f3f
#define N 310
using namespace std;
int G[N][N], s[N], n;
int visx[N], visy[N], use[N];
int lx[N], ly[N];
int Find(int u) //查询u是否存在完备匹配
{
    int i;
    visx[u] = 1;
    for (i = 1; i <= n; i++)
    {
        if (!visy[i] && G[u][i] == lx[u]+ly[i]) //当村庄i没有被查询，且u到i的权值刚好是最大权值
        {
            visy[i] = 1;
            if (!use[i] || Find(use[i])) //如果村庄i还有被匹配或者与i的匹配的点还能与其它点进行匹配
            {
                use[i] = u; //则可以将与i匹配的点换成u
                return 1; //代表u存在完备匹配
            }
        }
        else s[i] = min(s[i], (lx[u]+ly[i])-G[u][i]); //只有当G[u][i]越大是得到的s[i]越小，才能让权值更新范围d变小，继续下一次查询
    }
    return 0;
}
int KM()
{
    int i, j, ans = 0, d;
    memset(lx, 0, sizeof(lx));
    memset(ly, 0, sizeof(ly));
    for (i = 1; i <= n; i++)
    {
        for (j = 1; j <= n; j++)
            lx[i] = max(lx[i], G[i][j]);
    } //为了便于比较权值和更大的匹配，要保证（lx[i]+ly[j]）>=G[i][j]
    for (i = 1; i <= n; i++)    //那么lx数组可以初始化为与i相连的最大权值，ly数组初始化为0
    {
        while (1) //只要村民i没有找到能与之形成完备匹配的村庄j，就一直执行循环
        {
            memset(visx, 0, sizeof(visx));
            memset(visy, 0, sizeof(visy)); //每一次查找是否有完备匹配，都是从头开始，每次必须初始化
            for (j = 1; j <= n; j++)
                s[j] = INF;
            if (Find(i))
                break; //当找到完备匹配时就可以跳出循坏
            d = INF;
            for (j = 1; j <= n; j++)
            {
                if (!visy[j])
                    d = min(d, s[j]); //更新d的值，让它最小
            }
            for (j = 1; j <= n; j++)
            {
                if (visx[j]) lx[j] -= d; 
                if (visy[j]) ly[j] += d; //扩充相等子图（相等子图：为完备匹配中所有的匹配， 即所有的村民i和与之相匹配的村庄j）
            }
        }
    }
    for (i = 1; i <= n; i++)
        ans += G[use[i]][i]; //每次加上完备匹配的权值就是最大权值
    return ans;
}
int main ()
{
    int i, j, ans;
    while (scanf("%d", &n) != EOF)
    {
        memset(G, 0, sizeof(G));
        memset(use, 0, sizeof(use));
        for (i = 1; i <= n; i++)
        {
            for (j = 1; j <= n; j++)
                scanf("%d", &G[i][j]);
        }
        ans = KM();
        printf("%d\n", ans);
    }
    return 0;
}