HDU2255 KM优化(n^3)模板

转自：http://blog.youkuaiyun.com/cnyali/article/details/50663663

[cpp]  view plain  copy

1. #include <iostream>  
2. #include<cstring>  
3. #include<cstdio>  
4. #include<cmath>  
5. const int maxn = 101;  
6. const int INF = (1<<31)-1;  
7. int w[maxn][maxn];  
8. int lx[maxn],ly[maxn]; //顶标  
9. int match[maxn];  
10. int visx[maxn],visy[maxn];  
11. int slack[maxn];  
12. int nx,ny;  
13. bool find(int x)  
14. {  
15.     visx[x] = true;  
16.     for(int y = 0; y < ny; y++)  
17.     {  
18.         if(visy[y])  
19.             continue;  
20.         int t = lx[x] + ly[y] - w[x][y];  
21.         if(t==0)  
22.         {  
23.             visy[y] = true;  
24.             if(match[y]==-1 || find(match[y]))  
25.             {  
26.                 match[y] = x;  
27.                 return true;        //找到增广轨  
28.             }  
29.         }  
30.         else if(slack[y] > t)  
31.             slack[y] = t;  
32.     }  
33.     return false;                   //没有找到增广轨（说明顶点x没有对应的匹配，与完备匹配(相等子图的完备匹配)不符）  
34. }  
35.   
36. int KM()                //返回最优匹配的值  
37. {  
38.     int i,j;  
39.   
40.     memset(match,-1,sizeof(match));  
41.     memset(ly,0,sizeof(ly));  
42.     for(i = 0; i < nx; i++)  
43.         for(j = 0,lx[i] = -INF; j < ny; j++)  
44.             if(w[i][j] > lx[i])  
45.                 lx[i] = w[i][j];  
46.     for(int x = 0; x < nx; x++)  
47.     {  
48.         for(i = 0; i < ny; i++)  
49.             slack[i] = INF;  
50.         while(true)  
51.         {  
52.             memset(visx,0,sizeof(visx));  
53.             memset(visy,0,sizeof(visy));  
54.             if(find(x))                     //找到增广轨，退出  
55.                 break;  
56.             int d = INF;  
57.             for(i = 0; i < ny; i++)          //没找到，对l做调整(这会增加相等子图的边)，重新找  
58.             {  
59.                 if(!visy[i] && d > slack[i])  
60.                     d = slack[i];  
61.             }  
62.             for(i = 0; i < nx; i++)  
63.             {  
64.                 if(visx[i])  
65.                     lx[i] -= d;  
66.             }  
67.             for(i = 0; i < ny; i++)  
68.             {  
69.                 if(visy[i])  
70.                      ly[i] += d;  
71.                 else  
72.                      slack[i] -= d;  
73.             }  
74.         }  
75.     }  
76.     int result = 0;  
77.     for(i = 0; i < ny; i++)  
78.     if(match[i]>-1)  
79.         result += w[match[i]][i];  
80.     return result;  
81. }  
82.   
83. int main()  
84. {  
85.    // freopen("g:/1.txt","r",stdin);  
86.     while(true)  
87.     {  
88.         scanf("%d%d",&nx,&ny);  
89.         int a,b,c;  
90.         while(scanf("%d%d%d",&a,&b,&c),a+b+c)  
91.         {  
92.             w[a][b]=c;  
93.         }  
94.         printf("%d\n",KM());  
95.         break;  
96.     }  
97.     return 0;  
98. }