很巧妙的建图方法。
一开始只想到将单个颜色的做法,对于多个颜色就没想出来怎么建图了。
大体思路还是二分图最大独立集的方法。
首先黑白染色,分成黑白两个集合。设金的价值为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;
}