poj 3686 The Windy's

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

  1 #include <cstdio>

  2 #include <cstring>

  3 #include <algorithm>

  4 #define maxn 30000

  5 using namespace std;

  6 

  7 const int inf=1<<30;

  8 int weight[51][maxn],lx[maxn],ly[maxn],match[maxn],a[51][51];

  9 bool sx[maxn],sy[maxn];

 10 

 11 int n,m;

 12 

 13 bool path(int u)

 14 {

 15     sx[u]=true;

 16     for(int v=1; v<=n*m; v++)

 17     {

 18         if(!sy[v]&&lx[u]+ly[v]==weight[u][v])

 19         {

 20             sy[v]=true;

 21             if(!match[v]||path(match[v]))

 22             {

 23                 match[v]=u;

 24                 return true;

 25             }

 26         }

 27     }

 28     return false;

 29 }

 30 

 31 int bestmatch()

 32 {

 33     memset(lx,-inf,sizeof(lx));

 34     memset(ly,0,sizeof(ly));

 35     memset(match,0,sizeof(match));

 36     for(int i=1; i<=n; i++)

 37     {

 38         for(int j=1; j<=n*m; j++)

 39         {

 40             lx[i]=max(lx[i],weight[i][j]);

 41         }

 42     }

 43     for(int u=1; u<=n; u++)

 44     {

 45         while(1)

 46         {

 47             memset(sx,false,sizeof(sx));

 48             memset(sy,false,sizeof(sy));

 49             if(path(u)) break;

 50             int dx=inf;

 51             for(int i=1; i<=n; i++)

 52             {

 53                 if(sx[i])

 54                 {

 55                     for(int j=1; j<=n*m; j++)

 56                     {

 57                         if(!sy[j]) dx=min(lx[i]+ly[j]-weight[i][j],dx);

 58                     }

 59                 }

 60             }

 61             for(int i=1; i<=n; i++)

 62             {

 63                 if(sx[i]) lx[i]-=dx;

 64             }

 65             for(int j=1; j<=n*m; j++)

 66             {

 67                 if(sy[j]) ly[j]+=dx;

 68             }

 69         }

 70 

 71     }

 72     int sum=0;

 73     for(int i=1; i<=n*m; i++)

 74             sum+=weight[match[i]][i];

 75     return -sum;

 76 }

 77 int main()

 78 {

 79     int T;

 80     scanf("%d",&T);

 81     while(T--)

 82     {

 83         scanf("%d%d",&n,&m);

 84         for(int i=1; i<=n; i++)

 85         {

 86             for(int j=1; j<=m; j++)

 87             {

 88                 scanf("%d",&a[i][j]);

 89             }

 90         }

 91         for(int i=1; i<=n; i++)

 92         {

 93             for(int j=1; j<=n; j++)

 94             {

 95                 for(int k=1; k<=m; k++)

 96                 {

 97                     weight[i][(k-1)*n+j]=-a[i][k]*j;

 98                 }

 99             }

100         }

101         printf("%.6f\n",1.0*bestmatch()/n);

102     }

103     return 0;

104 }

View Code