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;
}