输入数据包含多组测试用例,每组数据的第一行输入n,表示男生的数量,接下来有n行,每行n个数表示第i个女生对第j个男生的好感度(n<=300) |
我们用fill函数来给slack赋极大值
#include<iostream>//slack:匹配
#include<algorithm>//match:松弛
#include<cstring>//gap:差距
#include<cstdio>
#define inf 0x3f3f3f3f
using namespace std;
int n;
int love[301][301]; ///记录每个女生对男生的好感度
int vis_girl[301],vis_boy[301];///记录是否匹配过女生,男生
int ex_girl[301],ex_boy[301];///记录女生,男生的期望值
int match[301],slack[301];///记录每个男生匹配的女生,如果没有就-1,记录每个男生如果能被女生倾心最少还需要多少期望值
int dfs(int x)
{
vis_girl[x]=1;
for(int i=0;i<n;i++)
{
if(vis_boy[i]) continue;///每一轮匹配,男生只尝试一次
int gap=ex_girl[x]+ex_boy[i]-love[x][i];
if(!gap){///如果符合要求
vis_boy[i]=1;
if(match[i]==-1||dfs(match[i])){///找到可以匹配的男生||该女生的男朋友可以找其他人
match[i]=x;
return 1;
}
}
else slack[i]=min(slack[i],gap);/// slack 可以理解为该男生要得到女生的倾心 还需多少期望值 取最小值 备胎的样子【捂脸
}
return 0;
}
void km()
{
memset(match,-1,sizeof(match));///所以男生都没女朋友
memset(ex_boy,0,sizeof(ex_boy));///每个男生的期望值都为零
for(int i=0;i<n;i++)///女生的期望值为她对男生的最大好感度
{
ex_girl[i]=love[i][0];
for(int j=1;j<n;j++)
ex_girl[i]=max(ex_girl[i],love[i][j]);
}
for(int i=0;i<n;i++)///尝试为每个女生找男朋友
{
fill(slack,slack+n,inf);///用fill给slack赋值
while(1){ ///寻找方法:找不到就降低期望值,直到找到为止
memset(vis_boy,0,sizeof(vis_boy)); ///重置访问记录
memset(vis_girl,0,sizeof(vis_girl));
if(dfs(i)) break; ///找到归宿就结束
int d=inf;
for(int j=0;j<n;j++) ///寻找最低需要降低的期望值
if(!vis_boy[j]) d=min(d,slack[j]);
for(int j=0;j<n;j++){
if(vis_girl[j]) ex_girl[j]-=d;///访问过的所以女生降低期望值
if(vis_boy[j]) ex_boy[j]+=d;///访问过的所以男生降低期望值
else slack[j]-=d;
}
}
}
int res=0;
for(int i=0;i<n;i++)
res+=love[match[i]][i];
printf("%d\n",res);
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
scanf("%d",&love[i][j]);
km();
}
return 0;
}
#include<iostream>//slack:匹配
#include<algorithm>//match:松弛
#include<cstring>//gap:差距
#include<cstdio>
#define inf 0x3f3f3f3f
using namespace std;
int n;
int love[301][301];
int vis_girl[301],vis_boy[301];
int ex_girl[301],ex_boy[301];
int match[301],slack[301];
int dfs(int x)
{
vis_girl[x]=1;
for(int i=0;i<n;i++)
{
if(vis_boy[i]) continue;
int gap=ex_girl[x]+ex_boy[i]-love[x][i];
if(!gap){
vis_boy[i]=1;
if(match[i]==-1||dfs(match[i])){
match[i]=x;
return 1;
}
}
else slack[i]=min(slack[i],gap);
}
return 0;
}
void km()
{
memset(match,-1,sizeof(match));
memset(ex_boy,0,sizeof(ex_boy));
for(int i=0;i<n;i++)
{
ex_girl[i]=love[i][0];
for(int j=1;j<n;j++)
ex_girl[i]=max(ex_girl[i],love[i][j]);
}
for(int i=0;i<n;i++)
{
fill(slack,slack+n,inf);
while(1){
memset(vis_boy,0,sizeof(vis_boy));
memset(vis_girl,0,sizeof(vis_girl));
if(dfs(i)) break;
int d=inf;
for(int j=0;j<n;j++)
if(!vis_boy[j]) d=min(d,slack[j]);
for(int j=0;j<n;j++){
if(vis_girl[j]) ex_girl[j]-=d;
if(vis_boy[j]) ex_boy[j]+=d;
else slack[j]-=d;
}
}
}
int res=0;
for(int i=0;i<n;i++)
res+=love[match[i]][i];
printf("%d\n",res);
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
scanf("%d",&love[i][j]);
km();
}
return 0;
}
下面这道km算法题需要自己构图: hdu2255(Chocolate,km算法+构图)