题目链接:http://www.lydsy.com:808/JudgeOnline/problem.php?id=3118
题意:给出一个图以及图中指定的n-1条边组成的生成树。每条边权值加1或者减去1都有相应的代价。求一个最小代价使得给出的边是最小生成树。
思路:对于每条非树边,必与某些树边形成环。设该非树边的权值为w2,某树边的权值为 w1。最后非树边增加x2,树边减少x1,那么w1-x1<=w2+x2。这样我们可以得到一些式子。代价也知道,这样就转化成线性规划问题。题目求的是最小值,我们可以将目标方程的系数取反求最大值。
单纯形的步骤:
(1)求出一个初始解;
(2)迭代。
(1)这个题的系数矩阵A是全么模:1、元素都是0,-1,1;2、任意子方阵的行列式为0,-1,1。
(2)据说A是全么模时解是整数解,因此此题可直接用单纯形。
const int COL=1005; const int ROW=30005; int n,m,B[ROW],N[COL]; double A[ROW][COL],b[ROW],c[COL],v; double ans[COL]; int sgn(double x) { if(x>1e-8) return 1; if(x<-1e-8) return -1; return 0; } //B中第l个替换N中第e个 void pivot(int l,int e) { int i,j; double temp=A[l][e]; b[l]/=temp; A[l][e]=1/temp; for(i=1;i<=n;i++) if(i!=e) A[l][i]/=temp; for(i=1;i<=m;i++) if(i!=l) { b[i]-=A[i][e]*b[l]; for(j=1;j<=n;j++) if(j!=e) A[i][j]-=A[i][e]*A[l][j]; A[i][e]=-A[i][e]/temp; } v+=b[l]*c[e]; for(i=1;i<=n;i++) if(i!=e) c[i]-=c[e]*A[l][i]; c[e]*=-A[l][e]; swap(B[l],N[e]); } void simplex() { int i,j,k,x; int l,s; double temp,temp1,temp2,temp3; while(1) { temp2=-dinf; s=-1; for(i=1;i<=n;i++) if(sgn(c[i])>0) { temp=dinf; for(k=1;k<=m;k++) if(sgn(A[k][i])>0) { temp3=b[k]/A[k][i]; if(temp3<temp) temp=temp3,x=k; } if(temp2<temp*c[i]) { s=i,l=x,temp2=temp*c[i]; } } if(s==-1) break; pivot(l,s); } for(i=1;i<=n;i++) { for(j=1;j<=m;j++) if(B[j]==i) break; if(j<=m) ans[i]=b[j]; else ans[i]=0; } } void print() { int i,j; printf("v: %.3lf\n",v); printf("c:\n"); for(i=1;i<=n;i++) printf("%.3lf ",c[i]); puts(""); printf("A:\n"); for(i=1;i<=m;i++) { for(j=1;j<=n;j++) printf("%.3lf ",A[i][j]); puts(""); } printf("b:\n"); for(i=1;i<=m;i++) printf("%.3lf ",b[i]); puts(""); printf("B:\n"); for(i=1;i<=m;i++) printf("%d ",B[i]); puts(""); printf("N:\n"); for(i=1;i<=n;i++) printf("%d ",N[i]); puts(""); } int init() { int i,j; int k=1; for(i=1;i<=m;i++) if(b[i]<b[k]) k=i; if(sgn(b[k])>=0) { for(i=1;i<=n;i++) N[i]=i; for(i=1;i<=m;i++) B[i]=n+i; v=0; simplex(); return 1; } static double tmpC[COL]; for(i=1;i<=n;i++) tmpC[i]=c[i]; tmpC[n+1]=0; n++; for(i=1;i<=m;i++) A[i][n]=-1; for(i=1;i<=n;i++) N[i]=i; for(i=1;i<=m;i++) B[i]=n+i; v=0; for(i=1;i<=n;i++) { if(i<n) c[i]=0; else c[i]=-1; } pivot(k,n); simplex(); if(sgn(ans[n])!=0) return 0; static int belongB[COL]; clr(belongB,0); for(i=1;i<=m;i++) { if(B[i]>n) continue; belongB[B[i]]=i; } map<int,int> mp; for(i=1;i<=n;i++) mp[N[i]]=i; clr(c,0); v=0; for(i=1;i<=n;i++) { if(!belongB[i]) { c[mp[i]]+=tmpC[i]; } else { v+=tmpC[i]*b[belongB[i]]; int j; for(j=1;j<=n;j++) { c[j]+=tmpC[i]*(-A[belongB[i]][j]); } } } c[mp[n]]=0; for(i=1;i<=m;i++) A[i][mp[n]]=0; simplex(); n--; return 1; } struct node { int u,v,id,w,next; }; node edges[ROW]; int e; int head[COL]; void add(int u,int v,int w,int id) { e++; edges[e].u=u; edges[e].v=v; edges[e].w=w; edges[e].id=id; edges[e].next=head[u]; head[u]=e; } int h[COL],up[COL],down[COL]; int eNum; int inq[COL],KK; int pre[COL]; void build(int s,int t,int p) { int i; for(i=pre[t];i!=-1;) { int w1=edges[i].w; int w2=edges[p*2].w; m++; A[m][edges[i].id]=-1; A[m][p]=-1; b[m]=w2-w1; int u=edges[i].u; i=pre[u]; } } void bfs(int s,int t,int p) { queue<int> Q; Q.push(s); KK++; inq[s]=KK; pre[s]=-1; while(!Q.empty()) { int u=Q.front(); Q.pop(); int i; for(i=head[u];i!=-1;i=edges[i].next) { int v=edges[i].v; int id=edges[i].id; if(!h[id]||KK==inq[v]) continue; pre[v]=i; if(v==t) { build(s,t,p); return; } Q.push(v); inq[v]=KK; } } } int main() { n=myInt(); eNum=myInt(); clr(head,-1); int i; for(i=1;i<=eNum;i++) { int u,v,w; scanf("%d%d%d%d%d%d",&u,&v,&w,&h[i],&up[i],&down[i]); add(u,v,w,i); add(v,u,w,i); } for(i=1;i<=eNum;i++) if(!h[i]) { int t=i*2; bfs(edges[t].u,edges[t].v,i); } for(i=1;i<=eNum;i++) c[i]=h[i]?-down[i]:-up[i]; n=eNum; if(!init()) { puts("no solution"); } double res=-v; if(res<1e-10) res=0; printf("%.0lf\n",-v); }
时间: 2024-11-13 06:37:37