题意:
给定n*n的矩阵,每个格子有个值s[i][j],现在要求对每行和每列各分配一个值,r[i],c[i]使得对所有的格子都有s[i][j]<=r[i]+c[j]成立,并且sum{r[i]}+sum{c[i]}尽量小。
直接KM啊。。SB题啊。。。
#include<iostream> #include<cstdio> #include<cstring> #include<string> #include<cstdlib> #include<cmath> #include<algorithm> #include<queue> #include<vector> #include<stack> using namespace std; #define INF 100000000 int n,nx,ny; int link[510],lx[510],ly[510],slack[510],w[510][510],visx[510],visy[510]; bool dfs(int x) { visx[x]=1; for(int y=1;y<=ny;y++) { if(visy[y]) continue; int t=lx[x]+ly[y]-w[x][y]; if(t==0) { visy[y]=1; if(link[y]==-1||dfs(link[y])) { link[y]=x; return 1; } } else if(slack[y]>t) slack[y]=t; } return 0; } void KM() { memset(link,-1,sizeof(link)); memset(ly,0,sizeof(ly)); for(int i=1;i<=nx;i++) { lx[i]=-INF; for(int j=1;j<=ny;j++) lx[i]=max(lx[i],w[i][j]); } for(int x=1;x<=nx;x++) { for(int i=1;i<=ny;i++) slack[i]=INF; while(1) { memset(visx,0,sizeof(visx)); memset(visy,0,sizeof(visy)); if(dfs(x)) break; int d=INF; for(int i=1;i<=ny;i++) { if(!visy[i]&&d>slack[i]) d=slack[i]; } for(int i=1;i<=nx;i++) { if(visx[i]) lx[i]-=d; } for(int i=1;i<=ny;i++) { if(visy[i]) ly[i]+=d; else slack[i]-=d; } } } int ans=0; for(int i=1;i<nx;i++) { ans+=lx[i]; printf("%d ",lx[i]); } ans+=lx[nx]; printf("%d\n",lx[nx]); for(int i=1;i<ny;i++) { ans+=ly[i]; printf("%d ",ly[i]); } ans+=ly[ny]; printf("%d\n",ly[ny]); printf("%d\n",ans); } int main() { while(scanf("%d",&n)!=EOF) { nx=ny=n; for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) scanf("%d",&w[i][j]); } KM(); } return 0; }
时间: 2024-10-13 16:52:04