二分图最大权匹配模板题HDU - 2255

题目链接：奔小康赚大钱

模板代码：

#include <bits/stdc++.h>
#define N 100010
#define mod 1000000007
#define INF 0x3f3f3f3f
using namespace std;

int w[301][301];
int p[305];
int h[305];
bool vis_p[305];
bool vis_h[305];
int match[305];
int slack[305];//记录每个房屋被村民购买至少需要多少期望值

int n;

bool dfs(int u)
{
    vis_p[u]=true;
    for(int i=0;i<n;i++)
    {
        if(vis_h[i]) continue;

        int gap=p[u]+h[i]-w[u][i];
        if(gap==0)//符合要求
        {
            vis_h[i]=true;
            if(match[i]==-1||dfs(match[i]))
            {
                match[i]=u;
                return 1;
            }
        }
        else
        {
            slack[i]=min(slack[i],gap);
        }
    }

    return false;

}

int KM()
{
    memset(match,-1,sizeof(match));
    memset(h,0,sizeof(h));
    for(int i=0;i<n;i++)
    {
        p[i]=w[i][0];
        for(int j=1;j<n;j++)
        {
            p[i]=max(p[i],w[i][j]);
        }
    }

    for(int i=0;i<n;i++)
    {
        memset(slack,INF,sizeof(slack));
        while(1)
        {
            memset(vis_h,false,sizeof(vis_h));
            memset(vis_p,false,sizeof(vis_p));
            if(dfs(i))
            {
                break;
            }//匹配成功，退出
            //如果未找到，就降低期望值
            int d=INF;
            for(int j=0;j<n;j++)
            {
                if(!vis_h[j])
                {
                    d=min(d,slack[j]);
                }
            }
            for(int j=0;j<n;j++)
            {
                if(vis_p[j])
                {
                    p[j]-=d;
                }
                if(vis_h[j])
                {
                    h[j]+=d;
                }
                else//没有访问过的房屋，期望值降低d
                {
                    slack[j]-=d;
                }
            }
        }
    }
    int res=0;
    for(int i=0;i<n;i++)
    {
        res+=w[match[i]][i];
    }
    return res;
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                scanf("%d",&w[i][j]);
            }
        }
        printf("%d\n",KM());
    }
    return 0;
}