[BZOJ 2127] happiness 【最小割】

题目链接:BZOJ - 2127

题目分析

首先,每个人要么学文科,要么学理科,所以可以想到是一个最小割模型。

我们就确定一个人如果和 S 相连就是学文,如果和 T 相连就是学理。

那么我们再来确定建图。首先使用最小割,就是先加上所有可能获得的权值,再减去最小割(即不能获得的权值)。

如果一个人学理,就要割掉与 S 相连的边,那么就是要割掉学文的收益。于是,对于每个点,从 S 向它连边,权值为它学文的收益。

同理,对于每个点,从它向 T 连边,权值为它学理的收益。

对于两个相邻的人,他们有同时学文的收益和同时学理的收益。

如果他们都学文,就会失去同时学理的收益,于是从他们分别向 T 连边,权值为他们同时学理收益的一半。

同理,从 S 向他们分别连边,权值为他们同时学文收益的一半。

但是如果一个人学文一个人学理,就要失去同时学文和同时学理的收益,因此,在他们之前连双向边,权值为同时学文的收益加同时学理的收益的一半。

这样建图,就巧妙地实现了控制不同的收益。

由于建图的时候权值的一半可能不是整数,先将权值乘 2 ,最后再除以 2 。

代码

#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm> using namespace std; const int MaxN = 100 + 5, MaxNode = 10000 + 15, INF = 999999999; int n, m, Sum, S, T, Tot, MaxFlow;
int Idx[MaxN][MaxN], Wk[MaxN][MaxN][5], Lk[MaxN][MaxN][5], Num[MaxNode], d[MaxNode]; struct Edge
{
int v, w;
Edge *Next, *Other;
} E[MaxNode * 12], *P = E, *Point[MaxNode], *Last[MaxNode]; inline void AddEdge(int x, int y, int z)
{
Edge *Q = ++P; ++P;
P -> v = y; P -> w = z;
P -> Next = Point[x]; Point[x] = P; P -> Other = Q;
Q -> v = x; Q -> w = 0;
Q -> Next = Point[y]; Point[y] = Q; Q -> Other = P;
} inline int gmin(int a, int b) {return a < b ? a : b;} int DFS(int Now, int Flow)
{
if (Now == T) return Flow;
int ret = 0;
for (Edge *j = Last[Now]; j; j = j -> Next)
if (j -> w && d[Now] == d[j -> v] + 1)
{
Last[Now] = j;
int p = DFS(j -> v, gmin(j -> w, Flow - ret));
ret += p; j -> w -= p; j -> Other -> w += p;
if (ret == Flow) return ret;
}
if (d[S] >= Tot) return ret;
if (--Num[d[Now]] == 0) d[S] = Tot;
++Num[++d[Now]];
Last[Now] = Point[Now];
return ret;
} int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
Idx[i][j] = (i - 1) * m + j;
S = n * m + 1; T = n * m + 2; Tot = T;
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
scanf("%d", &Wk[i][j][0]);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
scanf("%d", &Lk[i][j][0]);
for (int i = 1; i <= n - 1; ++i)
for (int j = 1; j <= m; ++j)
scanf("%d", &Wk[i][j][1]);
for (int i = 1; i <= n - 1; ++i)
for (int j = 1; j <= m; ++j)
scanf("%d", &Lk[i][j][1]);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m - 1; ++j)
scanf("%d", &Wk[i][j][2]);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m - 1; ++j)
scanf("%d", &Lk[i][j][2]);
int v1, v2, v3;
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
{
Sum += Wk[i][j][0] + Wk[i][j][1] + Wk[i][j][2];
Sum += Lk[i][j][0] + Lk[i][j][1] + Lk[i][j][2];
v1 = Wk[i][j][0] * 2 + Wk[i][j][1] + Wk[i][j][2];
v2 = Lk[i][j][0] * 2 + Lk[i][j][1] + Lk[i][j][2];
v1 += Wk[i - 1][j][1]; v2 += Lk[i - 1][j][1];
v1 += Wk[i][j - 1][2]; v2 += Lk[i][j - 1][2];
AddEdge(S, Idx[i][j], v1);
AddEdge(Idx[i][j], T, v2);
if (i < n)
{
v3 = Wk[i][j][1] + Lk[i][j][1];
AddEdge(Idx[i][j], Idx[i + 1][j], v3);
AddEdge(Idx[i + 1][j], Idx[i][j], v3);
}
if (j < m)
{
v3 = Wk[i][j][2] + Lk[i][j][2];
AddEdge(Idx[i][j], Idx[i][j + 1], v3);
AddEdge(Idx[i][j + 1], Idx[i][j], v3);
}
}
MaxFlow = 0;
memset(d, 0, sizeof(d));
memset(Num, 0, sizeof(Num)); Num[0] = Tot;
for (int i = 1; i <= Tot; ++i) Last[i] = Point[i];
while (d[S] < Tot) MaxFlow += DFS(S, INF);
Sum -= MaxFlow >> 1;
printf("%d\n", Sum);
return 0;
}

  

上一篇:ABP理论学习之开篇介绍


下一篇:Git 版本退回commit