您的位置:首页 > 其它

KM(n^4)模板

2016-02-14 17:36 281 查看
#include<stdio.h>
const int maxn=301;
int lx[maxn],ly[maxn],a[maxn][maxn],vx[maxn],vy[maxn],match[maxn],n;
bool dfs(int x){
vx[x]=1;
for(int i=1;i<=n;i++)
if(!vy[i] && lx[x]+ly[i]==a[x][i]){
vy[i]=1;
if(!match[i] || dfs(match[i])){
match[i]=x;
return 1;
}
}
return 0;
}
int main(){
int i,j,k,m;
scanf("%d",&n);
for(i=1;i<=n;i++){
int max=0;
for(j=1;j<=n;j++){
scanf("%d",&a[i][j]);
max=max>a[i][j]?max:a[i][j];
}
lx[i]=max;
}
for(i=1;i<=n;i++){
while(1){
for(j=1;j<=n;j++){vx[j]=0;vy[j]=0;}
if(dfs(i))break;
int min=100000000;
for(j=1;j<=n;j++)
if(vx[j])
for(k=1;k<=n;k++)
if(!vy[j])
min=min>(lx[j]+ly[k]-a[j][k])?min:lx[j]+ly[k]-a[j][k];
if(min==100000000){
printf("-1\n");
return 0;
}
for(j=1;j<=n;j++)
if(vx[j])lx[j]-=min;
for(j=1;j<=n;j++)
if(vy[j])ly[j]+=min;
}
}
int ans=0;
for(i=1;i<=n;i++)ans+=a[match[i]][i];
printf("%d\n",ans);
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: