HDU 2255 奔小康发大财

时间:2022-08-26 06:19:18

传送门

Solution:

KM算法

关于KM算法有一篇极好的文档http://www.cse.ust.hk/~golin/COMP572/Notes/Matching.pdf

Implementation:

#include <cstdio>
#include <algorithm>
#include <cstring>
#include <climits>
using namespace std;
const int N(); int w[N][N];
int Lx[N], Ly[N], slack[N];
bool S[N], T[N];
int match[N];
int n; bool dfs(int u)
{
S[u]=true;
for(int v=; v<=n; v++)
{
if(T[v])
{
continue;
}
int tmp=Lx[u]+Ly[v]-w[u][v];
if(tmp==)
{
T[v]=true;
if(!match[v] || dfs(match[v]))
{
match[v]=u;
return true;
}
}
else
{
slack[v]=min(slack[v], tmp);
}
}
return false;
} void KM()
{
memset(match, , sizeof(match));
memset(Lx, 0x3f, sizeof(Lx));
memset(Ly, 0x3f, sizeof(Ly)); for(int i=; i<=n; i++) //phase
{ for(int i=; i<=n; i++)
{
slack[i]=INT_MAX; //error-prone
}
for(int a; ;)
{
memset(S, , sizeof(S));
memset(T, , sizeof(T));
if(dfs(i)) break;
a=INT_MAX;
for(int j=; j<=n; j++)
{
if(!T[j])
{
a=min(a, slack[j]);
}
}
for(int j=; j<=n; j++)
{
if(S[j])
{
Lx[j]-=a;
}
if(T[j])
{
Ly[j]+=a;
}
else
{
slack[j]-=a;
}
}
}
}
int res=;
for(int i=; i<=n; i++)
{
res+=w[match[i]][i];
}
printf("%d\n", res);
} int main()
{
for(; ~scanf("%d", &n); )
{
for(int i=; i<=n; i++)
{
for(int j=; j<=n; j++)
{
scanf("%d", w[i]+j);
}
}
KM();
}
}

Error-prone:

我把Lx, Ly, slack都初始化成0x3f3f3f3f,导致dfs中

slack[v]=min(slack[v], tmp);

失灵。