poj2400--Supervisor, Supervisee(KM算法)

po2400：题目链接

题目大意：n个老板，n个职工，每个老板有对职工的一个排名，每个职工有对老板的一个排名，排名靠前，表示满意度高，表示想去那个老板那工作或是想要某个职工，现在每个老板选择一个职工，要求最小的平均差。如果有多个的话，按字典序输出

最有的平均差 = ∑所有人距离最想要的人的差/(2*n)。

题目的描述写反了，先输入的是职工对老板的排名，然后是老板的。

对每个关系进行编号，排名第一的值为0，第二的为-1，，，老板i对于职工j的值，应该是两个人对对方的值的和，先用km算法找出最大值，也就是最优的关系，然后dfs遍历所有情况，找出最可能会出现的匹配，用已经求出的结构优化搜索

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std ;
int Map[20][20] , n ;
int link[20] , vis[20] , cnt ;
int lx[20] , ly[20] ;
int s[20] , t[20] , ans ;
int dfs(int u) {
    s[u] = 1 ;
    for(int i = 1 ; i <= n ; i++) {
        if( !t[i] && lx[u]+ly[i] == Map[u][i] ) {
            t[i] = 1 ;
            if( link[i] == -1 || dfs(link[i]) ){
                link[i] = u ;
                return 1 ;
            }
        }
    }
    return 0 ;
}
void out(int dept,int temp) {
    if( temp > ans ) return ;
    if( dept > n ) {
        if( temp != ans ) return ;
        printf("Best Pairing %d\n", ++cnt) ;
        for(int i = 1 ; i <= n ; i++)
            printf("Supervisor %d with Employee %d\n", i, link[i]) ;
        return ;
    }
    else{
        for(int i = 1 ; i <= n ; i++) {
            if( vis[i] ) continue ;
            vis[i] = 1 ;
            link[dept] = i ;
            out(dept+1,temp-Map[dept][i]) ;
            vis[i] = 0 ;
        }
    }
}
int main() {
    int N , step = 0 , i , j , x , min1 , k ;
    scanf("%d", &N) ;
    while( N-- ) {
        memset(Map,0,sizeof(Map)) ;
        scanf("%d", &n) ;
        for(i = 1 ; i <= n ; i++)
            for(j = n ; j >= 1 ; j--)
                scanf("%d", &x) , Map[x][i] += j-n ;
        for(i = 1 ; i <= n ; i++)
            for(j = n ; j >= 1 ; j--)
                scanf("%d", &x) , Map[i][x] += j-n ;
        for(i = 1 ; i <= n ; i++){
            lx[i] = -100 ; ly[i] = 0 ;
            for(j = 1 ; j <= n ; j++)
                lx[i] = max(lx[i],Map[i][j]) ;
        }
        memset(link,-1,sizeof(link)) ;
        for(k = 1 ; k <= n ; k++) {
            while( 1 ) {
                memset(s,0,sizeof(s)) ;
                memset(t,0,sizeof(t)) ;
                if( dfs(k) ) break ;
                min1 = 100 ;
                for(i = 1 ; i <= n ; i++) {
                    if( !s[i] ) continue ;
                    for(j = 1 ; j <= n ; j++) {
                        if( t[j] ) continue ;
                        min1 = min(min1,lx[i]+ly[j]-Map[i][j]) ;
                    }
                }
                for(i = 1 ; i <= n ; i++) {
                    if( s[i] ) lx[i] -= min1 ;
                    if( t[i] ) ly[i] += min1 ;
                }
            }
        }
        for(i = 1 , ans = 0 ; i <= n ; i++)
            if( link[i] > 0 )
            ans -= Map[link[i]][i] ;
        printf("Data Set %d, Best average difference: %.6f\n", ++step, 0.5*ans/n) ;
        cnt = 0 ;
        memset(vis,0,sizeof(vis)) ;
        out(1,0) ;
        printf("\n") ;
    }
    return 0 ;
}