poj 3686The Windy's KM算法经典题

题目：http://poj.org/problem?id=3686

有N个订单在M个机器上加工，每个订单只能在同一台机器上加工完毕，中途不能更换。有N*M的矩阵，表示第i个订单在第j个机器上加工的用时，输出所用时间（等待时间和加工时间）的最小平均值

这道题刚开始感觉挺容易的，建图，套模板，然后WA，后来调试，发现了BUG，但是发现不会做了。。。

对于在某台机器上加工的第i个订单，显然它的用时是它本身的加工时间加在这台机器上加工的前i-1个订单的用时，无法独立计算每个订单的用时，因此需要转变思路。

对于每个物品而言，除了计算它本身消耗的时间，同时加上后面的物品会因为它的制造而额外增加的时间。制造当前这个物品需要的时间是t，而后面还有k个物品要制造，那么就等同于这个物品一共消耗了t+k*t的时间。这样转化之后就会发现计算每个物品的制造时间就仅依赖于后面物品的个数

假设某台机器加工了k个订单，那么用时为t1 + (t1 + t2) + (t1 + t2 + t3)......(t1+...tk)  =  t1 * k + t2 * (k-1) + t3 * (k-2) +...+tk，于是把每个机器拆成N个点，1-N分别代表倒数第几个加工了这个订单，然后KM算法即可

#include <iostream>
#include <string>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cctype>
#include <vector>
using namespace std;

const int N = 3000;
const int INF = 0x3f3f3f3f;
int nx, ny;
int lx[N], ly[N], match[N], slack[N];
bool visx[N], visy[N];
int s[60][N], mpa[60][60];

bool hungary(int v)
{
    visx[v] = true;
    for(int i = 0; i < ny; i++)
    {
        if(visy[i]) continue;
        if(lx[v] + ly[i] == s[v][i])
        {
            visy[i] = true;
            if(match[i] == -1 || hungary(match[i]))
            {
                match[i] = v;
                return true;
            }
        }
        else slack[i] = min(slack[i], lx[v] + ly[i] - s[v][i]);
    }

    return false;
}

void km()
{
    memset(ly, 0, sizeof ly);
    for(int i = 0; i < nx; i++)
        lx[i] = -INF;
    for(int i = 0; i < nx; i++)
        for(int j = 0; j < ny; j++)
            lx[i] = max(lx[i], s[i][j]);
    for(int i = 0; i < nx; i++)
    {
        memset(slack, 0x3f, sizeof slack);
        while(true)
        {
            memset(visx, 0, sizeof visx);
            memset(visy, 0, sizeof visy);

            if(hungary(i)) break;
            else
            {
                int tmp = INF;
                for(int j = 0; j < ny; j++)
                    if(!visy[j]) tmp = min(tmp, slack[j]);
                for(int j = 0; j < nx; j++)
                    if(visx[j]) lx[j] -= tmp;
                for(int j = 0; j < ny; j++)
                    if(visy[j]) ly[j] += tmp;
                    else slack[j] -= tmp;
            }
        }
    }
}

int main()
{
    int t, n, m, cnt, cost;

    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &m);


        for(int i = 0; i < n; i++)
        {
            cnt = 0;
            for(int j = 0; j < m; j++)
            {
                scanf("%d", &cost);
                for(int k = 1; k <= n; k++)
                    s[i][cnt++] = -cost * k;
            }
        }

        nx = n, ny = n * m;
        memset(match, -1, sizeof match);
        km();

        int res = 0;
        for(int i = 0; i < ny; i++)
            if(match[i] != -1)
                res += -s[match[i]][i];

        printf("%.6f\n", 1.0 * res / n);
    }

    return 0;
}