HDU 2255 奔小康赚大钱 （KM算法）

直接输入w[i][j]边权值建图套模板就可以了

//#pragma comment(linker, "/STACK:102400000,102400000")
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long LL;
const long long mod = 1e9+7;
const int INF = 0x3f3f3f3f;
const int M = 305;

/*求最大权匹配
 若求最小全匹配,可将权值取相反数,结果取相反数*/
 
int nx,ny;    //两边的点数
int g[M][M];     //二分图描述
int link[M],lx[M],ly[M];   //y中各点匹配状态,x,y中的点编号
int slack[M];
bool visx[M],visy[M];

bool dfs(int x)
{
    visx[x]=1;
    for(int y=0; y<ny; y++)
    {
        if(visy[y])
            continue;
        int tmp=lx[x]+ly[y]-g[x][y];
        if(!tmp)
        {
            visy[y]=1;
            if(link[y]==-1 || dfs(link[y]))
            {
                link[y]=x;
                return 1;
            }
        }
        else if(slack[y]>tmp)  //不在相等子图中slack 取最小的
        {
            slack[y]=tmp;
        }
    }
    return 0;
}

int KM()
{
    memset(link,-1,sizeof(link));
    memset(ly,0,sizeof(ly));
    for(int i=0; i<nx; ++i)   //lx,ly为顶标，nx,ny分别为x点集y点集的个数
    {
        lx[i]=-INF;
        for(int j=0; j<ny; ++j) 
        {
            if(g[i][j]>lx[i])
            {
                lx[i]=g[i][j];
            }
        }
    }
    for(int x=0; x<nx; ++x)
    {
        for(int i=0; i<ny; ++i)
        {
            slack[i]=INF;
        }
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(x))             //若成功（找到了增广轨），则该点增广完成，进入下一个点的增广
                break;            //若失败（没有找到增广轨），则需要改变一些点的标号，使得图中可行边的数量增加。
                                 //方法为：将所有在增广轨中（就是在增广过程中遍历到）的X方点的标号全部减去一个常数d，
                                //所有在增广轨中的Y方点的标号全部加上一个常数d

            int d=INF;
            for(int i=0; i<ny; ++i)
            {
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            }
            for(int i=0; i<nx; ++i)
            {
                if(visx[i])
                    lx[i]-=d;
            }
            for(int i=0; i<ny; ++i) //修改顶标后，要把所有不在交错树中的Y顶点的slack值都减去d
            {
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
            }
        }
    }
    int res=0;
    for(int i=0; i<ny; ++i)
    {
        if(link[i]!=-1)
            res+=g[link[i]][i];
    }
    return res;
}

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