(KM) uva 11383

题意:

  给定n*n的矩阵,每个格子有个值s[i][j],现在要求对每行和每列各分配一个值,r[i],c[i]使得对所有的格子都有s[i][j]<=r[i]+c[j]成立,并且sum{r[i]}+sum{c[i]}尽量小。

 

直接KM啊。。SB题啊。。。

 

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
using namespace std;
#define INF 100000000
int n,nx,ny;
int link[510],lx[510],ly[510],slack[510],w[510][510],visx[510],visy[510];
bool dfs(int x)
{
    visx[x]=1;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y])
            continue;
        int t=lx[x]+ly[y]-w[x][y];
        if(t==0)
        {
            visy[y]=1;
            if(link[y]==-1||dfs(link[y]))
            {
                link[y]=x;
                return 1;
            }
        }
        else if(slack[y]>t)
                slack[y]=t;
    }
    return 0;
}
void KM()
{
    memset(link,-1,sizeof(link));
    memset(ly,0,sizeof(ly));
    for(int i=1;i<=nx;i++)
    {
        lx[i]=-INF;
        for(int j=1;j<=ny;j++)
            lx[i]=max(lx[i],w[i][j]);
    }
    for(int x=1;x<=nx;x++)
    {
        for(int i=1;i<=ny;i++)
            slack[i]=INF;
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(x))
                break;
            int d=INF;
            for(int i=1;i<=ny;i++)
            {
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            }
            for(int i=1;i<=nx;i++)
            {
                if(visx[i])
                    lx[i]-=d;
            }
            for(int i=1;i<=ny;i++)
            {
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
            }
        }
    }
    int ans=0;
    for(int i=1;i<nx;i++)
    {
        ans+=lx[i];
        printf("%d ",lx[i]);
    }
    ans+=lx[nx];
    printf("%d\n",lx[nx]);
    for(int i=1;i<ny;i++)
    {
        ans+=ly[i];
        printf("%d ",ly[i]);
    }
    ans+=ly[ny];
    printf("%d\n",ly[ny]);
    printf("%d\n",ans);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        nx=ny=n;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
                scanf("%d",&w[i][j]);
        }
        KM();
    }
    return 0;
}

  

posted @ 2015-05-27 10:49  waterfull  阅读(268)  评论(0编辑  收藏  举报