poj 3686

KM练习

玩具和厂匹配。

厂按时间顺序拆点，边权为在某个时刻进厂前的等待时间。

降维时注意

j*n+k；j*m+k k*m+j k*n+j 中有两是错的！

#include <iostream>
#include <cstdio>
#include <cstring>
#define MAX 3000
using namespace std;
int n,m;
int _map[1000][1000],fmap[3000][3000];
int visx[MAX],visy[MAX],lx[MAX],ly[MAX],link[MAX],d,slack[100020];
void build()
{
    for(int i=0;i<n;i++)
    {
        int cnt=0;
        for(int j=0;j<m;j++)
        {
            for(int k=0;k<n;k++)
            {
                fmap[i][cnt++]=-_map[i][j]*(k+1);
            }
        }
    }
}
int dfs(int k)
{
    visx[k]=1;
    for(int i=0;i<n*m;i++)
    {
        if(visy[i]||!fmap[k][i])
        {
            continue;
        }
        int t=lx[k]+ly[i]-fmap[k][i];
        if(!t)
        {
            visy[i]=1;
            if(link[i]==-1||dfs(link[i]))
            {
                link[i]=k;
                return 1;
            }
        }
        else
        {
            if(slack[i]>t)
            {
                slack[i]=t;
            }
        }
    }
    return 0;
}
void KM()
{
    memset(link,-1,sizeof link);
    memset(lx,0,sizeof lx);
    memset(ly,0,sizeof ly);
    for(int i=0;i<n;i++)
    {
        lx[i]=-9999999;
        for(int j=0;j<n*m;j++)
        {
            if(fmap[i][j]>lx[i])
            {
                lx[i]=fmap[i][j];
            }
        }
    }
    for(int k=0;k<n;k++)
    {
        for(int i=0;i<n*m;i++)
        {
            slack[i]=999999;
        }
        while(1)
        {
            memset(visx,0,sizeof visx);
            memset(visy,0,sizeof visy);
            d=999999;
            if(dfs(k))
            {
                break;
            }
            for(int i=0;i<n*m;i++)
            {
                if(d>slack[i]&&!visy[i])
                {
                    d=slack[i];
                }
            }
            for(int i=0;i<n;i++)
            {
                if(visx[i])
                {
                    lx[i]-=d;
                }
            }
            for(int i=0;i<m*n;i++)
            {
                if(visy[i])
                {
                    ly[i]+=d;
                }
                else
                {
                    slack[i]+=d;
                }
            }
        }
    }
}
int main()
{
//    freopen("in","r",stdin);
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                scanf("%d",&_map[i][j]);
            }
        }

        build();
        KM();
        double ans=0;
        for(int i=0;i<n*m;i++)
        {
            if(link[i]!=-1)
            {
                ans+=fmap[link[i]][i];
//                printf("fmap=%d\n",fmap[link[i]][i]);
            }
        }
        printf("%lf\n",-ans/(double)n);
    }
    return 0;
}

另

#include <iostream>
#include <cstdio>
#include <cstring>
#define MAX 3000
using namespace std;
int n,m;
int _map[1000][1000],fmap[3000][3000];
int visx[MAX],visy[MAX],lx[MAX],ly[MAX],link[MAX],d;
void build()
{
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            for(int k=0;k<n;k++)
            {
                fmap[i][j+k*m]=-_map[i][j]*(k+1);
            }
        }
    }
}
int dfs(int k)
{
    visx[k]=1;
    for(int i=0;i<n*m;i++)
    {
        if(visy[i])
        {
            continue;
        }
        int t=lx[k]+ly[i]-fmap[k][i];
        if(!t)
        {
            visy[i]=1;
            if(link[i]==-1||dfs(link[i]))
            {
                link[i]=k;
                return 1;
            }
        }
        else
        {
            if(d>t)
            {
                d=t;
            }
        }
    }
    return 0;
}
void KM()
{
    memset(link,-1,sizeof link);
    memset(lx,0,sizeof lx);
    memset(ly,0,sizeof ly);
    for(int k=0;k<n;k++)
    {
        while(1)
        {
            memset(visx,0,sizeof visx);
            memset(visy,0,sizeof visy);
            d=999999;
            if(dfs(k))
            {
                break;
            }
            for(int i=0;i<n;i++)
            {
                if(visx[i])
                {
                    lx[i]-=d;
                }
            }
            for(int i=0;i<m*n;i++)
            {
                if(visy[i])
                {
                    ly[i]+=d;
                }
            }
        }
    }
}
int main()
{
//    freopen("in","r",stdin);
    int t;
    cin>>t;
    while(t--)
    {
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                scanf("%d",&_map[i][j]);
            }
        }

        build();
        KM();
        double ans=0;
        for(int i=0;i<n*m;i++)
        {
            if(link[i]!=-1)
            {
                ans+=fmap[link[i]][i];
            }
        }
        printf("%lf\n",-ans/(double)n);
    }
    return 0;
}