bzoj 2127 happiness
内容
Description
高一一班的座位表是个n*m的矩阵,经过一个学期的相处,每个同学和前后左右相邻的同学互相成为了好朋友。这学期要分文理科了,每个同学对于选择文科与理科有着自己的喜悦值,而一对好朋友如果能同时选文科或者理科,那么他们又将收获一些喜悦值。作为计算机竞赛教练的scp大老板,想知道如何分配可以使得全班的喜悦值总和最大。
Input
第一行两个正整数n,m。接下来是六个矩阵第一个矩阵为n行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学选择文科获得的喜悦值。第二个矩阵为n行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学选择理科获得的喜悦值。第三个矩阵为n-1行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i+1行第j列的同学同时选择文科获得的额外喜悦值。第四个矩阵为n-1行m列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i+1行第j列的同学同时选择理科获得的额外喜悦值。第五个矩阵为n行m-1列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i行第j+1列的同学同时选择文科获得的额外喜悦值。第六个矩阵为n行m-1列 此矩阵的第i行第j列的数字表示座位在第i行第j列的同学与第i行第j+1列的同学同时选择理科获得的额外喜悦值。
Output
输出一个整数,表示喜悦值总和的最大值
Sample Input
1 2
1 1
100 110
1
1000
Sample Output
1210
【样例说明】
两人都选理,则获得100+110+1000的喜悦值。
【数据规模】
对于100%以内的数据,n,m<=100 所有喜悦值均为小于等于5000的非负整数
题解
对于每个人,先假设他们选了理科,那么拿文科-理科,看正负连S/T,按照最大权闭合子图的方式建图。
对于每一个双人组新建一个点,根据文理连到S/T
一个点是指全部选文科,另一个是指有一个选了文科就要付出“共同选理科的代价”
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define N 60010
using namespace std;
const int inf = 0x3f3f3f3f;
struct flows {
struct node {
int to, next, flow;
} e[N * 10];
int tot, st[N], dis[N], cur[N];
void init() {
tot = -1, memset(e, -1, sizeof e),
memset(st, -1, sizeof st);
}
void add(int x, int y, int z) {
e[++tot].to = y;
e[tot].flow = z;
e[tot].next = st[x];
st[x] = tot;
}
void add_edge(int x, int y, int z) {
// printf("%d %d %d\n", x, y, z);
add(x, y, z), add(y, x, 0);
}
queue <int> que;
int bfs(int S, int T) {
memcpy(cur, st, sizeof st);
memset(dis, 0, sizeof dis);
while (!que.empty()) que.pop();
que.push(S);
dis[S] = 1;
while(!que.empty()) {
int now = que.front();
que.pop();
for (int i = st[now]; ~i; i = e[i].next)
if (e[i].flow && !dis[e[i].to]) {
dis[e[i].to] = dis[now] + 1;
if (e[i].to == T) return 1;
que.push(e[i].to);
}
}
return 0;
}
int finds(int now, int T, int flow) {
if (now == T)
return flow;
int f;
for (int i = cur[now]; ~i; i = e[i].next) {
cur[now] = i;
if (e[i].flow && dis[e[i].to] == dis[now] + 1 &&
(f = finds(e[i].to, T, min(flow, e[i].flow)))) {
e[i].flow -= f;
e[i^1].flow += f;
return f;
}
}
return 0;
}
int dinic(int S, int T) {
int flow = 0, x;
while(bfs(S, T)) {
while(x = finds(S, T, inf)) {
flow += x;
}
}
return flow;
}
}flow;
int n, m, a[110][110], x, all;
int ge(int x, int y) { return (x - 1) * m + y; }
main() {
scanf("%d%d", &n, &m);
int S = 0, T = n * m * 5 + 1;
flow.init();
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j)
scanf("%d", &a[i][j]);
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= m; ++j) {
scanf("%d", &x), all += x;
if (a[i][j] - x < 0)
flow.add_edge(ge(i, j), T, -(a[i][j] - x));
else
flow.add_edge(S, ge(i, j), a[i][j] - x), all += a[i][j] - x;
}
for (int i = 1; i < n; ++i)
for (int j = 1; j <= m; ++j) {
scanf("%d", &x), all += x;
flow.add_edge(S, n * m + ge(i, j), x);
flow.add_edge(n * m + ge(i, j), ge(i, j), inf);
flow.add_edge(n * m + ge(i, j), ge(i + 1, j), inf);
}
for (int i = 1; i < n; ++i)
for (int j = 1; j <= m; ++j) {
scanf("%d", &x), all += x;
flow.add_edge(n * m * 2 + ge(i, j), T, x),
flow.add_edge(ge(i, j), n * m * 2 + ge(i, j), inf),
flow.add_edge(ge(i + 1, j), n * m * 2 + ge(i, j), inf);
}
for (int i = 1; i <= n; ++i)
for (int j = 1; j < m; ++j) {
scanf("%d", &x), all += x;
flow.add_edge(S, n * m * 3 + ge(i, j), x),
flow.add_edge(n * m * 3 + ge(i, j), ge(i, j), inf),
flow.add_edge(n * m * 3 + ge(i, j), ge(i, j + 1), inf);
}
for (int i = 1; i <= n; ++i)
for (int j = 1; j < m; ++j) {
scanf("%d", &x), all += x;
flow.add_edge(n * m * 4 + ge(i, j), T, x),
flow.add_edge(ge(i, j), n * m * 4 + ge(i, j), inf),
flow.add_edge(ge(i, j + 1), n * m * 4 + ge(i, j), inf);
}
printf("%d\n", all - flow.dinic(S, T));
}