KM算法学习的是某牛的代码:http://www.cnblogs.com/Lyush/archive/2013/04/16/3025145.html
中间对slack的调整有个不懂的地方,但删去之后,还是能过,比源码慢了30ms左右。
加速原理不懂..
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 300
#define INF 0x3f3f3f3f
int slack[MAXN];
int Sx[MAXN],Sy[MAXN];
int W[MAXN][MAXN];
int Lx[MAXN],Ly[MAXN];
int Link[MAXN];
int n;
bool Find(int u)
{
Sx[u]=true;
for(int v=1;v<=n;v++)
{
if(Sy[v]) continue;
if(Lx[u]+Ly[v]==W[u][v])
{
Sy[v]=1;
if(!Link[v]||Find(Link[v]))
{
Link[v]=u;
return true;
}
}
else slack[v]=min(slack[v],Lx[u]+Ly[v]-W[u][v]);
}
return false;
}
int KM()
{
memset(Lx,0x80,sizeof(Lx));
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
Lx[i]=max(Lx[i],W[i][j]);
}
}
memset(Ly,0,sizeof(Ly));
memset(Link,0,sizeof(Link));
for(int i=1;i<=n;i++)
{
memset(slack,0x3f,sizeof(slack));
while(true)
{
memset(Sx,0,sizeof(Sx));
memset(Sy,0,sizeof(Sx));
if(Find(i)) break;
int d=INF;
for(int j=1;j<=n;j++)
{
if(!Sy[j]) d=min(d,slack[j]);
}
for(int j=1;j<=n;j++)
{
if(Sx[j]) Lx[j]-=d;
if(Sy[j]) Ly[j]+=d;
//else slack[j]-=d; 取消注释,可以加速KM算法,原理本弱不懂...
}
}
}
int ans=0;
for(int i=1;i<=n;i++) ans+=W[Link[i]][i];
return ans;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
int m;
scanf("%d%d",&n,&m);
memset(W,0x80,sizeof(W));
int from,to,val;
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&from,&to,&val);
W[from][to]=max(W[from][to],-val);
}
printf("%d\n",-KM());
}
return 0;
}