HDU - 2853 Assignment--KM优先使用原匹配边

题意

给出n个人和m个任务，一个人只能做一个任务，且一个任务只能被一个人做，给出人与任务之间做关系的价值矩阵，且给出每个人初始时所做的任务，先要求重新安排，求最大价值与原先价值的差值和最小改动任务分配。

思路

要求优先使用原匹配边，那么让边权*100+（若是原来使用的边？1:0），最后答案再/100就好了。 之所以选择*100，是因为若完全选择原先的边，也不会影响他们之间大小的比较。最大价值为ans/100，最小改动次数为n-ans%100。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define maxn 100+66
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define mod 1000000007
#define INF 0x3f3f3f3f

struct KM
{
    int nx,ny;
    int linker[maxn];
    int slack[maxn];
    int lx[maxn],ly[maxn];
    int visx[maxn],visy[maxn];
    int w[maxn][maxn];
    void init(int x,int y)
    {
        nx=x;
        ny=y;
        memset(w,INF,sizeof(w));
    }
    int dfs(int x)
    {
        visx[x]=1;
        for(int y=1; y<=ny; y++)
        {
            if(visy[y])
                continue;
            int tmp=lx[x]+ly[y]-w[x][y];
            if(tmp==0)
            {
                visy[y]=1;
                if(linker[y]==-1||dfs(linker[y]))
                {
                    linker[y]=x;
                    return 1;
                }
            }
            else if(slack[y]>tmp)
                slack[y]=tmp;
        }
        return 0;
    }
    int km()
    {
        int i,j;
        memset(linker,-1,sizeof(linker));
        memset(ly,0,sizeof(ly));
        for(i=1; i<=nx; i++)
            for(j=1,lx[i]=-INF; j<=ny; j++)
                if(w[i][j]>lx[i])
                    lx[i]=w[i][j];
        for(int x=1; x<=nx; x++)
        {
            for(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(i=1; i<=ny; i++)
                    if(!visy[i]&&d>slack[i])
                        d=slack[i];
                for(i=1; i<=nx; i++)
                    if(visx[i])
                        lx[i]-=d;
                for(i=1; i<=ny; i++)
                    if(visy[i])
                        ly[i]+=d;
                    else
                        slack[i]-=d;
            }
        }
        int res=0;
        for(i=1; i<=ny; i++)
            if(linker[i]!=-1)
                res+=w[linker[i]][i];
        return res;
    }
} KK;
int a[maxn][maxn],b[maxn][maxn];
int main()
{
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        rep(i,1,n)
        {
            rep(j,1,m)
            {
                scanf("%d",&a[i][j]);
            }
        }
        int pre=0;
        KK.init(n,m);
        rep(i,1,n)
        {
            int tmp;
            scanf("%d",&tmp);
            pre+=a[i][tmp];
            rep(j,1,m)
            {
                KK.w[i][j]=100*a[i][j]+(j==tmp?1:0);
            }
        }
        int ans=KK.km();
        int ans1=n-ans%100;
        int ans2=ans/100-pre;
        printf("%d %d\n",ans1,ans2);
    }
    return 0;
}