HDU 2255 奔小康赚大钱(二分图最优匹配:模板题)

HDU 2255 奔小康赚大钱(二分图最优匹配:模板题)

http://acm.hdu.edu.cn/showproblem.php?pid=2255

题意:

       给你一个带权的二分图，要求你求该二分图的最优匹配权值。

分析:

       左边点集代表村名，右边点集代表房子。本题就是KM模板题，直接用KM模板即可，不再赘述。

AC代码:

#include
#include
#include
using namespace std;
const int maxn=300+10;

struct Max_Match
{
    int n,W[maxn][maxn];
    int Lx[maxn],Ly[maxn];
    bool S[maxn],T[maxn];
    int left[maxn];

    bool match(int i)
    {
        S[i]=true;
        for(int j=1;j<=n;j++)if(Lx[i]+Ly[j]==W[i][j] && !T[j])
        {
            T[j]=true;
            if(left[j]==-1 || match(left[j]))
            {
                left[j]=i;
                return true;
            }
        }
        return false;
    }

    void update()
    {
        int a=1<<30;
        for(int i=1;i<=n;i++)if(S[i])
        for(int j=1;j<=n;j++)if(!T[j])
        {
            a=min(a, Lx[i]+Ly[j]-W[i][j]);
        }
        for(int i=1;i<=n;i++)
        {
            if(S[i]) Lx[i] -=a;
            if(T[i]) Ly[i] +=a;
        }
    }

    int solve(int n)
    {
        this->n=n;
        memset(left,-1,sizeof(left));
        for(int i=1;i<=n;i++)
        {
            Lx[i]=Ly[i]=0;
            for(int j=1;j<=n;j++)
                Lx[i]=max(Lx[i], W[i][j]);
        }
        for(int i=1;i<=n;i++)
        {
            while(true)
            {
                for(int j=1;j<=n;j++) S[j]=T[j]=false;
                if(match(i)) break;
                else update();
            }
        }
        int ans=0;
        for(int i=1;i<=n;i++) ans+= W[left[i]][i];
        return ans;
    }
}KM;


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