【hdu3820】【最小割】Golden Eggs

来源:互联网 发布:mac mpv 安装 编辑:程序博客网 时间:2024/05/18 20:49

很巧妙的建图方法。

一开始只想到将单个颜色的做法,对于多个颜色就没想出来怎么建图了。

大体思路还是二分图最大独立集的方法。

首先黑白染色,分成黑白两个集合。设金的价值为val1,银的价值为val2

由于有两种颜色,考虑将每个点拆成两个点k,k'

对于白色集合的点,k代表金,k'代表银;对于黑色集合的点,k代表银,k'代表金

白色集合的点:

s -> k,流量val1 ; k' -> t,流量val2

k -> k' ,流量为inf

黑色集合的点:

s -> k,流量为val2 ; k' -> t,流量val1

k -> k',流量为inf

每个白色集合的点k,与每个黑色集合的点 k' 连边,容量为g

每个黑色集合的点k , 与每个白色集合的点 k' 连边, 容量为s

最终答案为总价值 - 最小割

代码:

#include<cstdio>#include<cstring>using namespace std;const int maxg = 50 + 10;const int maxn = 5000 + 10;const int maxm = 200000;const int inf = 0x3f3f3f3f;struct Edge{int pos,c;int next;}E[maxm];int map1[maxg][maxg],map2[maxg][maxg],head[maxn];int dis[maxn],pre[maxn],cur[maxn],gap[maxn];int T,n,m,g,h;int NE,s,t,nodenum;int cas = 0,sum = 0;void init(){freopen("hdu3820.in","r",stdin);freopen("hdu3820.out","w",stdout);}void add(int u,int v,int c){E[NE].pos = v;E[NE].c = c;E[NE].next = head[u];head[u] = NE++;E[NE].pos = u;E[NE].c = 0;E[NE].next = head[v];head[v] = NE++;}void build_map(){memset(E,0,sizeof(E));memset(head,-1,sizeof(head));NE = 0;s = 0,t = n * m * 2 + 1;nodenum = t + 1;int p = n * m;for(int i = 1;i <= n;i++){for(int j = 1;j <= m;j++){add((i - 1) * m + j,(i - 1) * m + j + p,inf);if((i + j) & 1){add(s,(i - 1) * m + j,map2[i][j]);add((i - 1) * m + j + p,t,map1[i][j]);if(i > 1)add((i - 1) * m + j,(i - 2) * m + j + p,h);if(i < n)add((i - 1) * m + j,i * m + j + p,h);if(j > 1)add((i - 1) * m + j,(i - 1) * m + j + p - 1,h);if(j < m)add((i - 1) * m + j,(i - 1) * m + j + p + 1,h);}else{add(s,(i - 1) * m + j,map1[i][j]);add((i - 1) * m + j + p,t,map2[i][j]);if(i > 1)add((i - 1) * m + j,(i - 2) * m + j + p,g);if(i < n)add((i - 1) * m + j,i * m + j + p,g);if(j > 1)add((i - 1) * m + j,(i - 1) * m + j + p - 1,g);if(j < m)add((i - 1) * m + j,(i - 1) * m + j + p + 1,g);}}}}void checkmin(int &a,int b){if(a == -1 || a > b)a = b;}int sap(){memset(gap,0,sizeof(gap));memset(dis,0,sizeof(dis));for(int i = s;i <= t;i++)cur[i] = head[i];int u = pre[s] = s,maxflow = 0,aug = -1;gap[s] = nodenum;while(dis[s] < nodenum){loop:   for(int &i = cur[u];i != -1;i = E[i].next){int v = E[i].pos;if(E[i].c && dis[u] == dis[v] + 1){checkmin(aug,E[i].c);pre[v] = u;u = v;if(v == t){maxflow += aug;for(u = pre[u];v != s;v = u,u = pre[u]){E[cur[u]].c -= aug;E[cur[u]^1].c += aug;}aug = -1;}goto loop;}}int mind = nodenum;for(int i = head[u];i != -1;i = E[i].next){int v = E[i].pos;if(E[i].c && (mind > dis[v])){cur[u] = i;mind = dis[v];}}if((--gap[dis[u]]) == 0)break;gap[dis[u] = mind + 1]++;u = pre[u];}return maxflow;}void solve(){build_map();printf("Case %d: %d\n",++cas,sum - sap());}void readdata(){scanf("%d",&T);while(T--){memset(map1,0,sizeof(map1));memset(map2,0,sizeof(map2));scanf("%d%d%d%d",&n,&m,&g,&h);sum = 0;for(int i = 1;i <= n;i++){for(int j = 1;j <= m;j++){scanf("%d",&map1[i][j]);sum += map1[i][j];}}for(int i = 1;i <= n;i++){for(int j = 1;j <= m;j++){scanf("%d",&map2[i][j]);sum += map2[i][j];}}solve();}}int main(){init();readdata();return 0;}


原创粉丝点击