poj 3686 The Windy's

http://poj.org/problem?id=3686

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 30000
using namespace std;

const int inf=1<<30;
int weight[51][maxn],lx[maxn],ly[maxn],match[maxn],a[51][51];
bool sx[maxn],sy[maxn];

int n,m;

bool path(int u)
{
    sx[u]=true;
    for(int v=1; v<=n*m; v++)
    {
        if(!sy[v]&&lx[u]+ly[v]==weight[u][v])
        {
            sy[v]=true;
            if(!match[v]||path(match[v]))
            {
                match[v]=u;
                return true;
            }
        }
    }
    return false;
}

int bestmatch()
{
    memset(lx,-inf,sizeof(lx));
    memset(ly,0,sizeof(ly));
    memset(match,0,sizeof(match));
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<=n*m; j++)
        {
            lx[i]=max(lx[i],weight[i][j]);
        }
    }
    for(int u=1; u<=n; u++)
    {
        while(1)
        {
            memset(sx,false,sizeof(sx));
            memset(sy,false,sizeof(sy));
            if(path(u)) break;
            int dx=inf;
            for(int i=1; i<=n; i++)
            {
                if(sx[i])
                {
                    for(int j=1; j<=n*m; j++)
                    {
                        if(!sy[j]) dx=min(lx[i]+ly[j]-weight[i][j],dx);
                    }
                }
            }
            for(int i=1; i<=n; i++)
            {
                if(sx[i]) lx[i]-=dx;
            }
            for(int j=1; j<=n*m; j++)
            {
                if(sy[j]) ly[j]+=dx;
            }
        }

    }
    int sum=0;
    for(int i=1; i<=n*m; i++)
            sum+=weight[match[i]][i];
    return -sum;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m; j++)
            {
                scanf("%d",&a[i][j]);
            }
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=n; j++)
            {
                for(int k=1; k<=m; k++)
                {
                    weight[i][(k-1)*n+j]=-a[i][k]*j;
                }
            }
        }
        printf("%.6f
",1.0*bestmatch()/n);
    }
    return 0;
}