标签:style algorithm -- inf 意义 while 限制 包含 超过
长者对小明施加了膜法,使得小明每天起床就像马丁的早晨一样。 今天小明早上醒来发现自己成了一位仓管员。仓库可以被描述为一个n?×?m的网格,在每个网格上有几个箱子(可能没有)。为了防止倾倒,每个网格上,箱子不应该堆放超过h个。为了满足要求,小明需要搬一些箱子,每一次,他可以把箱子从一个网格到相邻的网格(如果两个网格共享一条边)。 请计算为了满足要求,小明至少需要搬多少次。
第一行一个整数T,表示有T组测试数据。 接下来每组数据,第一行有3个整数n,?m,?h(1?≤?n,?m,?h?≤?50), 分别表示仓库的大小为n?×?m,h意义见描述。 接下来n行,每行包含m个整数Aij(0?≤?Aij?≤?50), 表示位置(i,?j)上最初的箱子个数。 数据保证∑Aij?≤?n?×?m?×?h
输出小明至少需要搬多少次。
1 2 3 5 1 2 9 6 5 4
5
题解:
建立超级源点S,S对每个方格连费用为0,容量为方格箱子个数的边。
每个方格对其四周的所有方格连容量为正无穷,费用为1的边。
然后建立超级汇点T,限制每个方格都只能堆放h个盒子,所有方格对T连费用为0,容量为h的边.
#include <iostream> #include <algorithm> #include <stdio.h> #include <string.h> #include <queue> using namespace std; int mp[100][100]; int dir[][2] = {{1,0},{0,1},{-1,0},{0,-1}}; const int INF = 999999999; const int N = 3000; const int M = 6000000; struct Edge{ int u,v,cap,cost,next; }edge[M]; int head[N],tot,low[N],pre[N]; int total; bool vis[N]; void addEdge(int u,int v,int cap,int cost,int &k){ edge[k].u=u,edge[k].v=v,edge[k].cap = cap,edge[k].cost = cost,edge[k].next = head[u],head[u] = k++; edge[k].u=v,edge[k].v=u,edge[k].cap = 0,edge[k].cost = -cost,edge[k].next = head[v],head[v] = k++; } void init(){ memset(head,-1,sizeof(head)); tot = 0; } bool spfa(int s,int t,int n){ memset(vis,false,sizeof(vis)); for(int i=0;i<=n;i++){ low[i] = (i==s)?0:INF; pre[i] = -1; } queue<int> q; q.push(s); while(!q.empty()){ int u = q.front(); q.pop(); vis[u] = false; for(int k=head[u];k!=-1;k=edge[k].next){ int v = edge[k].v; if(edge[k].cap>0&&low[v]>low[u]+edge[k].cost){ low[v] = low[u] + edge[k].cost; pre[v] = k; ///v为终点对应的边 if(!vis[v]){ vis[v] = true; q.push(v); } } } } if(pre[t]==-1) return false; return true; } int MCMF(int s,int t,int n){ int mincost = 0,minflow,flow=0; while(spfa(s,t,n)) { minflow=INF+1; for(int i=pre[t];i!=-1;i=pre[edge[i].u]) minflow=min(minflow,edge[i].cap); flow+=minflow; for(int i=pre[t];i!=-1;i=pre[edge[i].u]) { edge[i].cap-=minflow; edge[i^1].cap+=minflow; } mincost+=low[t]*minflow; } total=flow; return mincost; } int main() { int n,m,h,tcase; scanf("%d",&tcase); while(tcase--){ scanf("%d%d%d",&n,&m,&h); init(); memset(mp,0,sizeof(mp)); for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ scanf("%d",&mp[i][j]); } } int s = 0,t = n*m+5; for(int i=1;i<=n;i++){ for(int j=1;j<=m;j++){ if(mp[i][j]!=0) addEdge(s,(i-1)*n+j,mp[i][j],0,tot); addEdge((i-1)*n+j,t,h,0,tot); for(int k=0;k<4;k++){ int nexti = i+dir[k][0]; int nextj = j+dir[k][1]; if(nexti>=1&&nexti<=n&&nextj>=1&&nextj<=m){ //printf("%d %d %d\n",nexti,nextj,(nexti-1)*n+nextj); addEdge((i-1)*n+j,(nexti-1)*n+nextj,INF,1,tot); } } } } int mincost = MCMF(s,t,n*m+5); cout<<mincost<<endl; } return 0; }
标签:style algorithm -- inf 意义 while 限制 包含 超过
原文地址:http://www.cnblogs.com/liyinggang/p/7434622.html