题意:n个人n场宴会,每个宴会有白天和黑夜两种选择,要求一半人参加白天的宴会,一半人参加晚上的,且一个宴会只有一个人能参加,求最大带权匹配。(n<=20)
思路:枚举每个宴会选择白天还是晚上,最后跑KM匹配,这样会得到一个
C(20,10)∗n3
,考虑到KM匹配是一个点一个点添加进去,因此显然可以在dfs的过程中顺便寻找增广路。另外,我采用之前的KM算法并没有通过这道题,学习了精妙的KM算法非递归版,其复杂度很容易看出是
O(n3)
,原来的写法虽然在实践中跑得非常快,但是最坏时间复杂度很难看出。
非递归的核心就在于其实每次就是根据一个左点找到一个拥有最小slack值的未在增广路上出现过的右点,假如这个右点未被匹配,那么找到增广路;否则继续沿该右点匹配的左点增广。
#include<bits/stdc++.h>
using namespace std;
const int inf=1e9;
int n;
int a[2][22][22];
int g[22][22];
int ans;
int wl[22],wr[22],lk[22],bef[22],d[22];
bool used[22];
void dfs(int now,int day,int night)
{
if(now>n){ans=max(ans,wr[0]);return;}
int tpwl[22],tpwr[22],tplk[22];
for(int ty=0;ty<2;ty++)
{
if(day+ty>n/2||night+(ty^1)>n/2)continue;
for(int i=0;i<=n;i++)tpwl[i]=wl[i],tpwr[i]=wr[i],tplk[i]=lk[i];
for(int i=1;i<=n;i++)g[now][i]=-a[ty][i][now];
for(int i=0;i<=n;i++)used[i]=0,d[i]=inf;
lk[0]=now;
int r=0;
while(lk[r]!=-1)
{
used[r]=1;
int l=lk[r],nxtr;
int slack=inf;
for(int i=1;i<=n;i++)
{
if(!used[i])
{
int t=g[l][i]-wl[l]-wr[i];
if(t<d[i]){d[i]=t;bef[i]=r;}
if(d[i]<slack)
{
slack=d[i];
nxtr=i;
}
}
}
for(int i=0;i<=n;i++)
{
if(used[i])
{
wl[lk[i]]+=slack;
wr[i]-=slack;
}
else d[i]-=slack;//这里别忘了
}
r=nxtr;
}
while(r)
{
lk[r]=lk[bef[r]];
r=bef[r];
}
dfs(now+1,day+ty,night+(ty^1));
for(int i=0;i<=n;i++)wl[i]=tpwl[i],wr[i]=tpwr[i],lk[i]=tplk[i];
}
}
int main()
{
scanf("%d",&n);
for(int i=0;i<2;i++)
{
for(int j=1;j<=n;j++)
for(int k=1;k<=n;k++)
scanf("%d",&a[i][j][k]);
}
memset(lk,-1,sizeof(lk));
dfs(1,0,0);
printf("%d\n",ans);
}