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;}