KM算法模版

#include<iostream>
#include<cstring>
#define N 500
using namespace std;

int n,w[N][N];
int lx[N],ly[N],f[N];bool vx[N],vy[N];

int Search(int k){
    vx[k]=1;
    
    for(int i=1;i<=n;++i)
    if(!vy[i]&&lx[k]+ly[i]==w[k][i])
    {
      if(f[i]==-1||Search(f[i]))
      {f[i]=k;return 1;}       
                                    }
    return 0;
    }

long KM(int p){
     if(!p)
     {
       for(int i=1;i<=n;++i)
       for(int j=1;j<=n;++j)
       w[i][j]=-w[i][j];
           }
     
     for(int i=1;i<=n;++i)
     {
       lx[i]=-100000000;
       for(int j=1;j<=n;++j)
       if(lx[i]<w[i][j]) lx[i]=w[i][j];    
       ly[i]=0;
             }
     
     memset(f,-1,sizeof(f));
     for(int i=1;i<=n;++i)
     {
       while(1)
       {
         memset(vx,0,sizeof(vx));
         memset(vy,0,sizeof(vy));
         
         if(Search(i)) break;
         
         int mi=100000000;
         for(int j=1;j<=n;++j)
         if(vx[j])
         for(int k=1;k<=n;++k)
         if(!vy[k]&&lx[j]+ly[k]-w[j][k]<mi)
         mi=lx[j]+ly[k]-w[j][k];
         
         if(mi==0) {cout<<"NO ANSWER!"<<endl;return 0;}
         for(int j=1;j<=n;++j)
         {
           if(vx[j]) lx[j]-=mi;
           if(vy[j]) ly[j]+=mi;
                 }
               }
             }
     
     int sum=0;
     for(int i=1;i<=n;++i)
     if(f[i]>0)
     sum+=w[f[i]][i];
     
     if(!p) return -sum;
     return sum;  
     } 

int main()
{
    cin>>n;
    for(int i=1;i<=n;++i)
    for(int j=1;j<=n;++j)
    cin>>w[i][j];
    
    cout<<KM(1)<<endl;   //最大 
    cout<<KM(0)<<endl;   //最小
    system("pause");
    return 0; 
    
    }