结合例子来学模板
奔小康赚大钱 HDU - 2255
const int maxn=307;
const int INF=0x3f3f3f3f;
int love[maxn][maxn]; // 记录每个妹子和每个男生的好感度
int ex_girl[maxn]; // 每个妹子的期望值
int ex_boy[maxn]; // 每个男生的期望值
bool vis_girl[maxn]; // 记录每一轮匹配匹配过的女生
bool vis_boy[maxn]; // 记录每一轮匹配匹配过的男生
int match[maxn]; // 记录每个男生匹配到的妹子 如果没有则为-1
int slack[maxn]; // 记录每个汉子如果能被妹子倾心最少还需要多少期望值
int N;
bool dfs(int girl)
{
vis_girl[girl]=true;
for(int boy=0;boy<N;boy++){
//每一轮匹配 每个男生只尝试一次
if(vis_boy[boy])
continue;
int gap=ex_girl[girl]+ex_boy[boy]-love[girl][boy];
//如果符合要求
if(gap==0){
vis_boy[boy]=true;
// 找到一个没有匹配的男生 或者该男生的妹子可以找到其他人
if(match[boy]==-1||dfs(match[boy])){
match[boy]=girl;
return true;
}
}
//slack 可以理解为该男生要得到女生的倾心 还需多少期望值 取最小值备胎的样子
else if(slack[boy]>gap)
slack[boy]=gap;
}
return false;
}
int KM()
{
//初始每个男生都没有匹配的女生
memset(match,-1,sizeof(match));
//初始每个男生的期望值为0
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++)
if(love[i][j]>ex_girl[i])
ex_girl[i]=love[i][j];
}
//尝试为每一个女生解决归宿问题
for(int i=0;i<N;i++){
//因为要取最小值 初始化为无穷大
for(int j=0;j<=N;j++)
slack[j]=INF;
// 为每个女生解决归宿问题的方法是 :如果找不到就降低期望值,直到找到为止
while(true){
//记录每轮匹配中男生女生是否被尝试匹配过
memset(vis_girl,false,sizeof(vis_girl));
memset(vis_boy,false,sizeof(vis_boy));
//找到归宿
if(dfs(i))
break;
//如果不能找到就降低期望值
//最小可降低的期望值
int d=INF;
for(int j=0;j<N;j++)
if(!vis_boy[j]&&d>slack[j])
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;
//没有访问过的boy 因为girl们的期望值降低,距离得到女生倾心又进了一步!
else
slack[j]-=d;
}
}
}
// 匹配完成求出所有配对的好感度的和
int res=0;
for(int i=0;i<N;i++)
res+=love[match[i]][i];
return res;
}
int main()
{
while(~scanf("%d",&N)){
for(int i=0;i<N;i++)
for(int j=0;j<N;j++)
scanf("%d",&love[i][j]);
printf("%d\n", KM());
}
return 0;
}
拓展应用
Engaging with Loyal Customers Kattis - engaging
# define INF 0x3f3f3f3f
const int maxn=2e3+7;
int love[maxn][maxn];
int ex_girl[maxn],ex_boy[maxn];
bool vis_girl[maxn],vis_boy[maxn];
int match[maxn];
int slack[maxn];
int nx,ny;
vector<pair<int,int> >vec;
bool dfs(int girl)
{
vis_girl[girl]=true;
for(int boy=1;boy<=ny;boy++){
if(vis_boy[boy])
continue;
int gap=ex_girl[girl]+ex_boy[boy]-love[girl][boy];
if(gap==0){
vis_boy[boy]=true;
if(match[boy]==-1||dfs(match[boy])){
match[boy]=girl;
return true;
}
}
else
slack[boy]=min(slack[boy],gap);
}
return false;
}
pair<int,int> KM()
{
memset(match,-1,sizeof(match));
memset(ex_boy,0,sizeof(ex_boy));
for(int i=1;i<=nx;i++){
ex_girl[i]=-INF;
for(int j=1;j<=ny;j++)
ex_girl[i]=max(ex_girl[i],love[i][j]);
}
for(int i=1;i<=nx;i++){
for(int j=1;j<=ny;j++)
slack[j]=INF;
while(true){
memset(vis_girl,false,sizeof(vis_girl));
memset(vis_boy,false,sizeof(vis_boy));
if(dfs(i))
break;
int d=INF;
for(int j=1;j<=ny;j++)
if(!vis_boy[j])
d=min(d,slack[j]);
for(int j=1;j<=nx;j++){
if(vis_girl[j])
ex_girl[j]-=d;
}
for(int j=1;j<=ny;j++){
if(vis_boy[j])
ex_boy[j]+=d;
else
slack[j]-=d;
}
}
}
int sum=0,num=0;
for(int i=1;i<=ny;i++){
if(match[i]>0){
sum+=love[match[i]][i];
num++;
vec.push_back(make_pair(match[i],i));
}
}
return make_pair(sum,num);
}
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
bool flag_change=true;
nx=n;
ny=m;
if(n>m){
nx=m;
ny=n;
flag_change=false;
}
while(k--){
int customers,gift,level;
scanf("%d%d%d",&customers,&gift,&level);
if(!flag_change)
love[customers][gift]=level;
else
love[gift][customers]=level;
}
pair<int,int> temp=KM();
printf("%d\n%d\n",temp.first,temp.second);
int len=vec.size();
for(int i=0;i<len;i++){
if(flag_change)
printf("%d %d\n",vec[i].second,vec[i].first);
else
printf("%d %d\n",vec[i].first,vec[i].second);
}
return 0;
}