“玲珑杯”ACM比赛 Round #4 B Best couple

一眼的KM，但是建图的时候记得不用的点设为0，点少的一边补齐，这个非常重要，因为KM追求完全匹配，如果无法完全匹配会非常慢

#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
#define MAXN 105
#define _clr(x) memset(x,-1,sizeof(int)*MAXN)

int mp[MAXN][MAXN], match1[MAXN], match2[MAXN]; 
int KM(int m,int n,int mat[][MAXN],int *match1,int *match2)
{
    int s[MAXN],t[MAXN],l1[MAXN],l2[MAXN];
    int p,q,i,j,k,ret=0;
    for(i=0;i<m;i++)
    {
        l1[i]=-INF;
        for(j=0;j<n;j++)
            l1[i]=mat[i][j]>l1[i]?mat[i][j]:l1[i];
        if(l1[i]==-INF)  return -1;
    } 
    for(i=0;i<n;i++)
        l2[i]=0;
    _clr(match1);
    _clr(match2);
    for(i=0;i<m;i++)
    {
        _clr(t);
        p=0;q=0;
        for(s[0]=i;p<=q&&match1[i]<0;p++)
        {
            for(k=s[p],j=0;j<n&&match1[i]<0;j++)
            {
                if(l1[k]+l2[j]==mat[k][j]&&t[j]<0)
                {
                    s[++q]=match2[j];
                    t[j]=k;
                    if(s[q]<0)
                    {
                        for(p=j;p>=0;j=p)
                        {
                            match2[j]=k=t[j];
                            p=match1[k];
                            match1[k]=j;
                        }    
                    }    
                }    
            }    
        } 
        if(match1[i]<0)
        {
            i--;
            p=INF;
            for(k=0;k<=q;k++)
            {
                for(j=0;j<n;j++)
                {
                    if(t[j]<0&&l1[s[k]]+l2[j]-mat[s[k]][j]<p)
                        p=l1[s[k]]+l2[j]-mat[s[k]][j];
                }    
            }  
            for(j=0;j<n;j++)
                l2[j]+=t[j]<0?0:p;
            for(k=0;k<=q;k++)
                l1[s[k]]-=p;  
        }       
    } 
    for(i=0;i<m;i++)
        ret+=mat[i][match1[i]];
    return ret;      
}

int d[205][205];
int main(){
    int n,m;
    int _; scanf("%d",&_);
    while(_--) {
        scanf("%d %d",&n,&m);
        for(int i = 1; i <= n+m; ++i)
            for(int j = 1; j <= n+m; ++j) {
                scanf("%d",&d[i][j]);
                if(d[i][j] == -1) d[i][j] = INF;
            }
        for(int k = 1; k <= n+m; ++k) {
            for(int i = 1; i <= n+m; ++i) {
                for(int j = 1; j <= n+m; ++j) {
                    d[i][j] = min(d[i][j], d[i][k]+d[k][j]);
                }
            }
        }

        for(int i = 1; i <= n; ++i) {
            for(int j = 1; j <= m; ++j) {
                mp[i-1][j-1] = d[i][n+j];
                if(mp[i-1][j-1] == INF) mp[i-1][j-1] = 0;
            }
        }
        if(n > m) {
            for(int i = 1; i <= n; ++i) {
                for(int j = m+1; j <= n; ++j) {
                    mp[i-1][j-1] = 0;
                }
            }
        }else if(n < m) {
            for(int i = n+1; i <= m; ++i)
                for(int j = 1; j <= m; ++j) {
                    mp[i-1][j-1] = 0;
                }
        }

        int tt = max(n, m);
        printf("%d\n",KM(tt,tt,mp,match1,match2));
    }
    return 0;
}

转载于:https://www.cnblogs.com/Basasuya/p/8433725.html