滚动建图,最大费用流(每次只有就10个点的二分图)。复杂度,m/n*(n^2)(n<=10),今年网络赛唯一网络流题,被队友状压DP秒了。。。。难道网络流要逐渐退出历史舞台???。。。。
#include<iostream> //78ms #include<cstdio> #include<queue> using namespace std; const double inf =0x3f3f3f3f; const int maxv=50,maxe=500; int head[maxv];double e[maxe][4];int nume=0; void inline adde(int i,int j,int c,double w) { e[nume][0]=j;e[nume][1]=head[i];head[i]=nume; e[nume][2]=c;e[nume++][3]=w; e[nume][0]=i;e[nume][1]=head[j];head[j]=nume; e[nume][2]=0;e[nume++][3]=-w; } int n,m;int ss,tt; void init() { nume=0; for(int i=0;i<=2*n+2;i++) head[i]=-1; ss=0;tt=2*n+1; } int inq[maxv];double d[maxv];int prv[maxv];int pre[maxv]; bool spfa(double & sums) { for(int i=0;i<=tt;i++) { inq[i]=0; d[i]=inf; } queue<int>q; q.push(ss); inq[ss]=1; d[ss]=0; while(!q.empty()) { int cur=q.front(); q.pop(); inq[cur]=0; for(int j=head[cur];j!=-1;j=e[j][1]) { int v=e[j][0]; if(d[v]>d[cur]+e[j][3]&&e[j][2]>0) { d[v]=d[cur]+e[j][3]; pre[v]=j; prv[v]=cur; if(!inq[v]) { inq[v]=1; q.push(v); } } } } if(d[tt]==inf)return 0; double minf=inf; int cur=tt; while(cur!=ss) { if(minf>e[pre[cur]][2]) minf=e[pre[cur]][2]; cur=prv[cur]; } cur=tt; while(cur!=ss) { e[pre[cur]][2]-=minf; e[pre[cur]^1][2]+=minf; cur=prv[cur]; } sums+=minf*d[tt]; return 1; } double mincost() { double sums=0; while(spfa(sums)); return sums; } double a[11][1005]; int main() { int T; scanf("%d",&T);int cnt=1; while(T--) { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) scanf("%lf",&a[i][j]); double sums=0; for(int ii=0;ii<m/n;ii++) { init(); for(int i=1;i<=n;i++) { adde(ss,i,1,0); adde(i+n,tt,1,0); for(int j=n+1;j<=n+n;j++) { adde(i,j,1,-a[i][(j-n)+n*ii]); } } sums+=mincost(); } init(); for(int i=n+1;i<=n+(m%n);i++) adde(i,tt,1,0); for(int i=1;i<=n;i++) { adde(ss,i,1,0); for(int j=n+1;j<=n+(m%n);j++) { adde(i,j,1,-a[i][(j-n)+n*(m/n)]); } } sums+=mincost(); printf("Case #%d: %.5lf\n",cnt++,-sums); } return 0; }
时间: 2024-12-14 18:50:46