ZOJ 2539 Energy Minimization

把题目意思理解清楚以后，就会发现这道题跟POJ 3469 那道双核处理器是一样的

 

公式第一项只要当xi=0时才会有作用，第二项只有当xi=1时才会有作用，第三项只有当xi和xj不相等时才会有作用

 

对于每个点i,xi要么等于0，要么等于1，即点i要么属于S集，要么属于T集，如果点i,j不在同一个集合，它们之间还会产生附加值

 

总的最小value值正好对应一个最小割的容量，value=Sum(v0i)+Sum(v1j)+Sum(vij),令v0为源点，v1为汇点，则i表示点i属于S集，点j表示j属于T集，当(s,i)为割边时，(i,t)不会是割边，同理，（i,t）为割边时，(s,i)不会是割边，当i,j在同一个集合时，（i,j）不会是割边

 

这题公式第三项1<=i<j<=N,j属于N(i)，但下面说j属于N(i)表示的是点i的左右和上下邻点，但由于j>i，只能取右边邻点和下边邻点（如果存在的话），一开始在这里WA了几次

 

还有就是除了最后一组测试数据，每组结束后都要多打一行空行，一开始只打了一个换行符，在调试的时候看起来时空了一行，实际是那行本来要输出结果的，但有由于要先读数据，把那行空出来了，所以还要多打一个换行符，囧。。这个PE了两次

 

代码：

#include<iostream> #include<stdio.h> #include<stdlib.h> #include<math.h> #include<memory.h> using namespace std; const int inf=1<<30; const int MAX=505; struct node { int v,c,next; }g[MAX*MAX*2]; int dis[MAX],num[MAX],cur[MAX],pre[MAX],adj[MAX],map[MAX][MAX]; int s,t,e,vn; void add(int u,int v,int c) { g[e].v=v; g[e].c=c; g[e].next=adj[u]; adj[u]=e++; g[e].v=u; g[e].c=0; g[e].next=adj[v]; adj[v]=e++; } int ABS(int x) { return x>0?x:-x; } int sap() { int i,u,v,flow=0,aug=inf,flag; for(i=0;i<=vn;i++) { dis[i]=num[i]=0; cur[i]=adj[i]; } num[0]=vn; pre[s]=u=s; while(dis[s]<vn) { flag=0; for(i=cur[u];i!=-1;i=g[i].next) { v=g[i].v; if(g[i].c&&dis[u]==dis[v]+1) { flag=1; pre[v]=u; cur[u]=i; aug=aug<g[i].c?aug:g[i].c; u=v; if(u==t) { flow+=aug; while(u!=s) { u=pre[u]; g[cur[u]].c-=aug; g[cur[u]^1].c+=aug; } aug=inf; } break; } } if(flag) continue; if(--num[dis[u]]==0) break; for(dis[u]=vn,i=adj[u];i!=-1;i=g[i].next) { v=g[i].v; if(g[i].c&&dis[v]<dis[u]) { dis[u]=dis[v]; cur[u]=i; } } dis[u]++; num[dis[u]]++; u=pre[u]; } return flow; } int main() { int i,j,k,T,ca,v0,v1,r,col; scanf("%d",&T); for(ca=1;ca<=T;ca++) { scanf("%d %d %d %d",&r,&col,&v0,&v1); e=0; memset(adj,-1,sizeof(adj)); memset(map,0,sizeof(map)); s=0; t=r*col+1; vn=t+1; for(i=1;i<=r;i++) { for(j=1;j<=col;j++) { scanf("%d",&map[i][j]); } } for(i=1;i<=r;i++) { for(j=1;j<=col;j++) { k=(i-1)*col+j; add(s,k,ABS(map[i][j]-v0)); add(k,s,ABS(map[i][j]-v0)); add(k,t,ABS(map[i][j]-v1)); add(t,k,ABS(map[i][j]-v1)); /*if(i>1) { add(k,k-col,ABS(map[i][j]-map[i-1][j])); add(k-col,k,ABS(map[i][j]-map[i-1][j])); }*/ if(i<r) { add(k,k+col,ABS(map[i][j]-map[i+1][j])); add(k+col,k,ABS(map[i][j]-map[i+1][j])); } /*if(j>1) { add(k,k-1,ABS(map[i][j]-map[i][j-1])); add(k-1,k,ABS(map[i][j]-map[i][j-1])); }*/ if(j<col) { add(k,k+1,ABS(map[i][j]-map[i][j+1])); add(k+1,k,ABS(map[i][j]-map[i][j+1])); } } } printf("Case %d:/n",ca); printf("%d/n",sap()); if(ca<T) printf("/n"); } return 0; }