POJ 2400 Supervisor, Supervisee KM求最小权

共有N个管理者和N个雇员，每个雇员会跟随一个管理者。每个雇员会对管理者的喜好程度进行排名，管理者也同样会对雇员进行排名，问如何对管理者和雇员进行配对使总排名的平均值最小。

如果最喜欢，设为0，以此递减。

之后对于（i，j）建边两者相加的和。

求最小权，也就是匹配之后的最小值ans。

平均值就是ans/（2*n）。

之后要求输出所有的匹配情况。

就是dfs，当总值达到ans且所有老板都访问过了，那就输出每个老板对应的员工。之后继续dfs

1. #include <stdio.h>  
2. #include <string.h>  
3. const int inf=1<<30;  
4. int w[20][20],n,km1[20],km2[20],d,pre[20],cnt,ans;  
5. bool visx[20],visy[20];  
6. void init()  
7. {  
8.     memset(km2,0,sizeof(km2));  
9.     memset(pre,-1,sizeof(pre));  
10.     for(int i=1; i<=n; i++)  
11.     {  
12.         km1[i]=w[i][1];  
13.         for(int j=2; j<=n; j++)  
14.         {  
15.             if(km1[i]<w[i][j])km1[i]=w[i][j];  
16.         }  
17.     }  
18. }  
19. bool dfs(int src)  
20. {  
21.     visx[src]=true;  
22.     for(int i=1; i<=n; i++)  
23.     {  
24.         if(!visy[i])  
25.         {  
26.             int t=km1[src]+km2[i]-w[src][i];  
27.             if(!t)  
28.             {  
29.                 visy[i]=true;  
30.                 if(pre[i]==-1||dfs(pre[i]))  
31.                 {  
32.                     pre[i]=src;  
33.                     return true;  
34.                 }  
35.             }  
36.             else if(d>t)d=t;  
37.         }  
38.     }  
39.     return false;  
40. }  
41. int km()  
42. {  
43.     for(int i=1; i<=n; i++)  
44.     {  
45.         while(1)  
46.         {  
47.             memset(visx,false,sizeof(visx));  
48.             memset(visy,false,sizeof(visy));  
49.             d=inf;  
50.             if(dfs(i))break;  
51.             for(int j=1; j<=n; j++)  
52.             {  
53.                 if(visx[j])km1[j]-=d;  
54.                 if(visy[j])km2[j]+=d;  
55.             }  
56.         }  
57.     }  
58.     ans=0;  
59.     for(int i=1; i<=n; i++)  
60.     {  
61.         if(pre[i]!=-1)ans+=w[pre[i]][i];  
62.     }  
63.     return -ans;  
64. }  
65. void dfs(int src,int ca)  
66. {  
67.     if(ca<ans)return;  
68.     if(src>n)  
69.     {  
70.         if(ca!=ans)return;  
71.         printf("Best Pairing %d\n",cnt++);  
72.         for(int i=1; i<=n; i++)  
73.         {  
74.             printf("Supervisor %d with Employee %d\n",i,pre[i]);  
75.         }  
76.     }  
77.     else  
78.     {  
79.         for(int i=1; i<=n; i++)  
80.         {  
81.             if(!visx[i])  
82.             {  
83.                 visx[i]=true;  
84.                 pre[src]=i;  
85.                 dfs(src+1,ca+w[src][i]);  
86.                 visx[i]=false;  
87.             }  
88.         }  
89.     }  
90. }  
91. int main()  
92. {  
93.     int t,x,case1=1;  
94.     scanf("%d\n",&t);  
95.     while(t--)  
96.     {  
97.         scanf("%d\n",&n);  
98.         memset(w,0,sizeof(w));  
99.         for(int i=1; i<=n; i++)  
100.         {  
101.             for(int j=1; j<=n; j++)  
102.             {  
103.                 scanf("%d",&x);  
104.                 w[x][i]-=(j-1);  
105.             }  
106.         }  
107.         for(int i=1; i<=n; i++)  
108.         {  
109.             for(int j=1; j<=n; j++)  
110.             {  
111.                 scanf("%d",&x);  
112.                 w[i][x]-=(j-1);  
113.             }  
114.         }  
115.         init();  
116.         printf("Data Set %d, Best average difference: %lf\n",case1++,(km()*0.5)/n);  
117.         memset(visx,0,sizeof(visx));  
118.         cnt=1;  
119.         dfs(1,0);  
120.         printf("\n");  
121.     }  
122.     return 0;  
123. }