HDU-2255 奔小康賺大錢
阿新 • • 發佈:2018-11-02
二分圖最大權匹配模板題,用了KM演算法,也可用最小費用最大流
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<cmath> #include<queue> using namespace std; typedef long long ll; const int N=300+10; const int INF=0x7f7f7f7f; int nx,ny; int g[N][N]; int linker[N],lx[N],ly[N]; int slack[N]; bool visx[N],visy[N]; bool dfs(int x) { visx[x]=true; for(int y=0;y<ny;y++) { if(visy[y]) continue; int tmp=lx[x]+ly[y]-g[x][y]; if(tmp==0) { visy[y]=true; if(linker[y]==-1||dfs(linker[y])) { linker[y]=x; return true; } } else if(slack[y]>tmp) slack[y]=tmp; } return false; } int KM() { memset(linker,-1,sizeof(linker)); memset(ly,0,sizeof(ly)); for(int i=0;i<nx;i++) { lx[i]=-INF; for(int j=0;j<ny;j++) if(g[i][j]>lx[i]) lx[i]=g[i][j]; } for(int x=0;x<nx;x++) { for(int i=0;i<ny;i++) slack[i]=INF; while(true) { memset(visx,false,sizeof(visx)); memset(visy,false,sizeof(visy)); if(dfs(x)) break; int d=INF; for(int i=0;i<ny;i++) if(!visy[i]&&d>slack[i]) d=slack[i]; for(int i=0;i<nx;i++) if(visx[i]) lx[i]-=d; for(int i=0;i<ny;i++) { if(visy[i]) ly[i]+=d; else slack[i]-=d; } } } int res=0; for(int i=0;i<ny;i++) if(linker[i]!=-1) res+=g[linker[i]][i]; return res; } int main() { int n; while(~scanf("%d",&n)) { nx=ny=n; for(int i=0;i<n;i++) for(int j=0;j<n;j++) scanf("%d",&g[i][j]); printf("%d\n",KM()); } return 0; }