1076. Trash（KM算法 二分最佳完美匹配）

1076. Trash

Input

Output

Sample

4
62 41 86 94
73 58 11 12
69 93 89 88
81 40 69 13

650

#include<iostream>
#include<sstream>
#include<algorithm>
#include<cstdio>
#include<string.h>
#include<cctype>
#include<string>
#include<cmath>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
using namespace std;
const int INF=10000;
int W[INF][INF],n;
int Lx[INF],Ly[INF];//
int Left[INF];  //
bool S[INF],T[INF];//
int sum[INF];
int matrix[INF][INF];
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]||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  KM()
{
    for(int i=1; i<=n; i++)
    {
        Left[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++)
    {
        for(;;)
        {
            for(int j=1; j<=n; j++) S[j]=T[j]=0;
            if(match(i))break;
            else update();
        }
    }
    int sum = 0;
    for(int i = 1; i <= n; i ++)
        if(match[i] > 0)
            sum += W[Left[i]][i];
    return sum;
}

int main()
{
    while(cin>>n)
    {
        memset(sum,0,sizeof(sum));
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
            {
                scanf("%d",&matrix[i][j]);
                sum[j]+=matrix[i][j];
            }
        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
                W[i][j]=-(sum[j]-matrix[i][j]);
        cout<<-KM()<<endl;
    }
    return 0;
}