KM算法 PK 最小费用最大流

时间：2022-04-30 14:13:05

用到了KM算法 ,发现自己没有这个模板,搜索学习一下上海大学final大神,http://www.cnblogs.com/kuangbin/p/3228861.html

 #include <stdio.h>

 #include <algorithm>

 #include <string.h>

 #include <iostream>

 using namespace std;

 /*  KM算法

  *   复杂度O(nx*nx*ny)

  *  求最大权匹配

  *   若求最小权匹配，可将权值取相反数，结果取相反数

  *  点的编号从0开始

  */

 const int N = ;

 const int INF = 0x3f3f3f3f;

 int nx,ny;//两边的点数

 int g[N][N];//二分图描述

 int linker[N],lx[N],ly[N];//y中各点匹配状态，x,y中的点标号

 int slack[N];

 bool visx[N],visy[N];

 bool DFS(int x)

 {

     visx[x] = true;

     for(int y = ; y < ny; y++)

     {

         if(visy[y])continue;

         int tmp = lx[x] + ly[y] - g[x][y];

         if(tmp == )

         {

             visy[y] = true;

             if(linker[y] == - || DFS(linker[y]))

             {

                 linker[y] = x;

                 return true;

             }

         }

         else if(slack[y] > tmp)

             slack[y] = tmp;

     }

     return false;

 }

 int KM()

 {

     memset(linker,-,sizeof(linker));

     memset(ly,,sizeof(ly));

     for(int i = ;i < nx;i++)

     {

         lx[i] = -INF;

         for(int j = ;j < ny;j++)

             if(g[i][j] > lx[i])

                 lx[i] = g[i][j];

     }

     for(int x = ;x < nx;x++)

     {

         for(int i = ;i < ny;i++)

             slack[i] = INF;

         while(true)

         {

             memset(visx,false,sizeof(visx));

             memset(visy,false,sizeof(visy));

             if(DFS(x))break;

             int d = INF;

             for(int i = ;i < ny;i++)

                 if(!visy[i] && d > slack[i])

                     d = slack[i];

             for(int i = ;i < nx;i++)

                 if(visx[i])

                     lx[i] -= d;

             for(int i = ;i < ny;i++)

             {

                 if(visy[i])ly[i] += d;

                 else slack[i] -= d;

             }

         }

     }

     int res = ;

     for(int i = ;i < ny;i++)

         if(linker[i] != -)

             res += g[linker[i]][i];

     return res;

 }

 //HDU 2255

 int main()

 {

     int n;

     while(scanf("%d",&n) == )

     {

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

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

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

         nx = ny = n;

         printf("%d\n",KM());

     }

     return ;

 }

end

相关文章