Supervisor, Supervisee 【POJ - 2400】【KM算法】

题目链接

---

  题意：

  这道题的题意不是那么的明确，并且输入和阐述也有一些小冲突。具体是这样的：

  有N个老板和N个来面试的员工，每个老板都对应的有第一想要的员工～第N想要的员工（最不想要的员工），然后呢，员工也对应有最想去的老板公司单位，也分为1～N序列。现在呢，我们希望贡献最大，贡献是怎样算的呢，如果两方都是第一喜欢的话，那么贡献为0，有譬如说老板第x喜欢某员工，且该员工第y喜欢该老板，那么它们的贡献是(x + y - 2) == (x - 1 + y - 1)。现在啊，我们希望贡献最大，并且求得这个贡献的。

  最后，我们可能会有多种最优解的情况，分别按照升序来输出这些答案，也就是说老板的1～N是固定的，后面跟着的对应员工，分别按照下标升序来输出。

  思路：

  知道了这些之后就是比较的简单了，我们可以先求最小权（如果我们换成最大的话，剪枝会比较浪费了），也就是建立负边来做。然后我们由于N只有最多15，所以我们可以用dfs爆搜加一个小剪枝来完成这个搜索。直接输出即可。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
//#include <unordered_map>
//#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define eps 1e-10
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
#define Min3(a, b, c) min(a, min(b, c))
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int maxN = 16;
int N;
ll ans_val;
struct KM
{
    int n, mp[maxN][maxN], link_x[maxN], link_y[maxN];
    bool vis_x[maxN], vis_y[maxN];
    int que[maxN], top, fail, pre[maxN];
    ll hx[maxN], hy[maxN], slk[maxN];
    bool check(int i)
    {
        vis_x[i] = true;
        if(link_x[i])
        {
            que[fail++] = link_x[i];
            vis_y[link_x[i]] = true;
            return true;
        }
        while(i)
        {
            link_x[i] = pre[i];
            swap(i, link_y[pre[i]]);
        }
        return false;
    }
    void bfs(int S)
    {
        for(int i=1; i<=n; i++)
        {
            slk[i] = INF;
            vis_x[i] = vis_y[i] = false;
        }
        top = 0; fail = 1; que[0] = S;
        vis_y[S] = true;
        while(true)
        {
            ll d = 0;
            while(top < fail)
            {
                for(int i=1, j = que[top++]; i<=n; i++)
                {
                    if(!vis_x[i] && slk[i] >= (d = hx[i] + hy[j] - mp[i][j]))
                    {
                        pre[i] = j;
                        if(d) slk[i] = d;
                        else if(!check(i)) return;
                    }
                }
            }
            d = INF;
            for(int i=1; i<=n; i++)
            {
                if(!vis_x[i] && d > slk[i]) d = slk[i];
            }
            for(int i=1; i<=n; i++)
            {
                if(vis_x[i]) hx[i] += d;
                else slk[i] -= d;
                if(vis_y[i]) hy[i] -= d;
            }
            for(int i=1; i<=n; i++)
            {
                if(!vis_x[i] && !slk[i] && !check(i)) return;
            }
        }
    }
    void Init()
    {
        for(int i=1; i<=n; i++)
        {
            link_x[i] = link_y[i] = 0;
            hy[i] = slk[i] = 0;
            vis_y[i] = false;
        }
        for(int i=1; i<=n; i++)
        {
            hx[i] = 0;
            for(int j=1; j<=n; j++)
            {
                if(hx[i] < mp[i][j]) hx[i] = mp[i][j];
            }
        }
    }
    void solve()
    {
        n = N;
        Init();
        for(int i=1; i<=n; i++) bfs(i);
        ll ans = 0;
        for(int i=1; i<=n; i++) ans += hx[i] + hy[i];
        ans_val = -ans;
        printf("%.6f\n", eps - 1. * ans / (2 * N));
    }
}km;
bool used[maxN] = {false};
int choose[maxN], Best_id;
void dfs(int boss_id, ll now_val)
{
    if(now_val > ans_val) return;
    if(boss_id == N + 1)
    {
        printf("Best Pairing %d\n", ++Best_id);
        for(int i=1; i<=N; i++)
        {
            printf("Supervisor %d with Employee %d\n", i, choose[i]);
        }
        return;
    }
    for(int i=1; i<=N; i++)
    {
        if(used[i]) continue;
        used[i] = true;
        choose[boss_id] = i;
        dfs(boss_id + 1, now_val - km.mp[boss_id][i]);
        used[i] = false;
    }
}
int main()
{
    int T, Cas = 0; scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &N);
        Best_id = 0;
        for(int i=1, id; i<=N; i++) //员工
        {
            for(int j=0; j<N; j++)
            {
                scanf("%d", &id);
                km.mp[id][i] = j;
            }
        }
        for(int i=1, id; i<=N; i++) //领导
        {
            for(int j=0; j<N; j++)
            {
                scanf("%d", &id);
                km.mp[i][id] = km.mp[i][id] + j;
            }
        }
        for(int i=1; i<=N; i++) for(int j=1; j<=N; j++) km.mp[i][j] = -km.mp[i][j];
        printf("Data Set %d, Best average difference: ", ++Cas);
        km.solve();
        memset(used, false, sizeof(used));
        dfs(1, 0);
        printf("\n");
    }
    return 0;
}