POJ3686-KM匹配模板题

#include <cstdio>#include <cstring>#include <iostream>using namespace std;const int NN=2550;const int INF=0x3fffffff;int n,m,lx[NN],ly[NN],w[NN][NN],a[NN][NN],match[NN],slack[NN];bool vx[NN],vy[NN];void init(){    scanf("%d%d",&n,&m);    for (int i=0; i<n; i++)      for (int j=0; j<m; j++) scanf("%d",&a[i][j]);    for (int i=0; i<n; i++)      for (int j=0; j<m; j++)        for (int k=1; k<=n; k++)        {            w[i][j+(k-1)*m]=-a[i][j]*k;        }    m=n*m;}bool dfs(int u){    vx[u]=true;    for (int v=0; v<m; v++)    {        if (vy[v]) continue;        int t=lx[u]+ly[v]-w[u][v];        if (t==0)        {            vy[v]=true;            if (match[v]==-1 || dfs(match[v]))            {                match[v]=u;                return true;            }        }        else if (slack[v]>t) slack[v]=t;    }    return false;}double KM(){    memset(match,-1,sizeof(match));    memset(lx,0,sizeof(lx));    memset(ly,0,sizeof(ly));    for (int i=0; i<n; i++)        for (int j=0; j<m; j++) if (w[i][j]>lx[i]) lx[i]=w[i][j];    for (int x=0; x<n; x++)    {        for (int i=0; i<m; i++) slack[i]=INF;        while (1)        {            memset(vx,0,sizeof(vx));            memset(vy,0,sizeof(vy));            if (dfs(x)) break;            int d=INF;            for (int i=0; i<m; i++)                if (!vy[i] && d>slack[i]) d=slack[i];            for (int i=0; i<n; i++) if (vx[i]) lx[i]-=d;            for (int i=0; i<m; i++) if (vy[i]) ly[i]+=d; else slack[i]-=d;        }    }    double ret=0;    for (int i=0; i<m; i++)    {        if (match[i]!=-1) ret+=w[match[i]][i];    }    return ret;}int main(){    int cas;    scanf("%d",&cas);    while (cas--)    {        init();        printf("%.6f\n",-KM()/(n*1.0));    }    return 0;}