hdu2426 二分最优匹配KM算法

这题。。。。很明显的二分匹配。但是要注意没有匹配上的情况。

由于学生和宿舍时一一匹配，则需要n<m且n<（能匹配的宿舍）

同时，在求KM算法中要判断是否能完备匹配，即在找d的时候，判断是否能找到。

注意图的初始化为负无穷。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define CLR(arr, what) memset(arr, what, sizeof(arr))
#define maxn  550
#define INF  (1<<30)-1
int g[maxn][maxn];int flag[maxn];
int lx[maxn],ly[maxn],match[maxn];
bool visx[maxn],visy[maxn];
int slack[maxn];
int n,m,e;
bool dfs(int cur){
     visx[cur] = true;
     for(int y=1;y<=m;y++){
         if(visy[y])   continue;
         int t=lx[cur]+ly[y]-g[cur][y];
         if(t==0){
            visy[y] = true;
            if(match[y]==-1||dfs(match[y])){
                match[y] = cur;
                return true;
            }
         }
         else if(slack[y]>t){
                 slack[y]=t;
         }
     }
     return false;
}
int KM(){
    CLR(match,-1);
    CLR(ly,0);;
    for(int i=1 ;i<=n;i++){
         lx[i]=-INF;
       for(int j=1;j<=m;j++)
           if(g[i][j]>lx[i])   lx[i]=g[i][j];
   }
   for(int x=1;x<=n;x++){
        for(int i=1;i<=m;i++)  slack[i]=INF;
        while(true){
            CLR(visx,false);
            CLR(visy,false);
            if(dfs(x))  break;
            int d=INF;
            for(int i=1;i<=m;i++){
               if(!visy[i]&&d>slack[i])     d=slack[i];
            }
            if(d==INF)
                return -1;
            for(int i=1;i<=n;i++){
               if(visx[i])                  lx[i]-=d;
            }
            for(int i=1;i<=m;i++){
               if(visy[i])                 ly[i]+=d;
               else                        slack[i]-=d;
            }
        }
   }
    int result = 0;
    for(int i = 1; i <=m; i++)
    if(match[i]!=-1)
        result += g[match[i]][i];
    return result;
}
int main(){
    int cas=1;
    while(scanf("%d%d%d",&n,&m,&e)!=EOF){
        CLR(g,0);
        CLR(flag,0);
        int numx=0;
        for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            g[i][j]=-INF;
        for(int i=1;i<=e;i++)
        {
            int v,w,r;
            scanf("%d%d%d",&v,&w,&r);
            if(r>=0)
            {
                g[v+1][w+1]=r;
            }
                if(!flag[w+1])numx++;
                flag[w+1]=1;

        }
        printf("Case %d: ",cas++);
        if(numx<n||n>m)
        {
            printf("-1\n");
            continue;
        }
        printf("%d\n",KM());
    }
    return 0;
}