牛客网暑期ACM多校训练营（第五场） E(KM算法)

题目链接：https://www.nowcoder.com/acm/contest/143/E

 

题目大意：先给n个4组数据表示第一年n个寝室的人员分配，再来n个四组数据表示第二年哪些同学想在一起住，问最少调用几个同学可以让大家都满意

 

题目思路：KM裸题，建图直接根据两个寝室之间的相同人数作为权值，然后跑出全图的最大权匹配，表示最多有多少人不用换寝室，然后总人数4*n减去最大全匹配就是想要的答案，直接套模板就过了...

 

以下是代码：

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
#define ll long long
#define MAXN 105
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
int a[MAXN][5],b[MAXN][5],g[MAXN][MAXN];
int nx,ny;
int linker[MAXN],lx[MAXN],ly[MAXN];
int slack[MAXN];
bool visx[MAXN],visy[MAXN];
bool dfs(int x){
    visx[x]=true;
    rep(y,1,ny){
        if(visy[y])continue;
        int tmp=lx[x]+ly[y]-g[x][y];
        if(!tmp){
            visy[y]=true;
            if(linker[y]==-1||dfs(linker[y])){
                linker[y]=x;
                return true;
            }
        }
        else if(slack[y]>tmp){
            slack[y]=tmp;
        }
    }
    return false;
}
int km(){
    memset(linker,-1,sizeof(linker));
    memset(ly,0,sizeof(ly));
    rep(i,1,nx){
        lx[i]=-inf;
        rep(j,1,ny){
            if(g[i][j]>lx[i]){
                lx[i]=g[i][j];
            }
        }
    }
    rep(x,1,nx){
        rep(i,1,ny){
            slack[i]=inf;
        }
        while(true){
            memset(visx,false,sizeof(visx));
            memset(visy,false,sizeof(visy));
            if(dfs(x))break;
            int d=inf;
            rep(i,1,ny){
                if(!visy[i]&&d>slack[i])d=slack[i];
            }
            rep(i,1,nx){
                if(visx[i]){
                    lx[i]-=d;
                }
            }
            rep(i,1,ny){
                if(visy[i])ly[i]+=d;
                else slack[i]-=d;
            }
        }
    }
    int res=0;
    rep(i,1,ny){
        if(linker[i]!=-1){
            res+=g[linker[i]][i];
        }
    }
    return res;
}
int main()
{
    int n;
    while(~scanf("%d",&n)){
        rep(i,1,n){
            rep(j,1,4){
                scanf("%d",&a[i][j]);
            }
        }
        rep(i,1,n){
            rep(j,1,4){
                scanf("%d",&b[i][j]);
            }
        }
        nx=ny=n;
        rep(i,1,n){
            rep(j,1,n){
                int num=0;
                rep(k,1,4){
                    rep(p,1,4){
                        if(a[i][k]==b[j][p])num++;
                    }
                }
                g[i][j]=num;
            }
        }
        printf("%d\n",4*n-km());
    }
    return 0;
}