[BZOJ 3774] 最優選擇 【最小割】

題目連接：BZOJ - 3774

 

題目分析

此題與「文理分科」那道題目有些相似。都是使用最小割來求解，先加上可能得到的權值，在減掉必須捨棄的權值（最小割）。

文理分科是規定每一個人和 S 連就是選文，和 T 連就是選理。而後若是一我的和相鄰的人都全文就會得到一個權值，那麼咱們就爲這個權值建一個點，讓這個點與必須同時選文的5我的連 INF 邊。這樣只要這 5 我的中有一我的選了理，就必須捨棄這個權值了。

再回到這道題目，這道題得到權值的條件是這個點被控制或這個點相鄰的 4 個點都被控制。 這個「或」並不太好處理，咱們就把這個條件拆成兩個不相交的條件：

1)這個點被控制，能夠得到權值。

2)這個點沒有被控制且相鄰的4個點都被控制，能夠得到權值。

這樣的話第一個條件就是不控制這個點須要付出的代價，第二個條件是「這個點沒有被控制且相鄰的4個點都被控制」，只要有一個點不符合就要割掉這個權值。

可是這些須要同時知足的條件有「被控制」和「不被控制」，直接用「文理分科」的建圖方式是方向不一致的。

因此咱們利用矩陣能夠黑白染色成爲二分圖的性質，將矩陣的格子黑白染色以後，對於白點和黑點用相反的方式連邊，這樣黑點的「被控制」和白點的「不被控制」就是一個方向的了。

注意剛開始時預先加到答案裏的權值是給定權值的兩倍，由於咱們將權值分紅了兩種狀況。

 

代碼

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>

using namespace std;

inline int gmin(int a, int b) {return a < b ? a : b;}

const int MaxMap = 50 + 5, MaxN = 5000 + 5, MaxM = 100000 + 5, INF = 999999999;
const int Dx[5] = {0, 0, 1, -1}, Dy[5] = {1, -1, 0, 0};

int n, m, nm, Tot, Ans, S, T;
int f[MaxMap][MaxMap], A[MaxMap][MaxMap], B[MaxMap][MaxMap], d[MaxN], Num[MaxN];

struct Edge
{
	int v, w;
	Edge *Next, *Other;
} E[MaxM], *P = E, *Point[MaxN], *Last[MaxN];

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 bool Inside(int x, int y)
{
	if (x < 1 || x > n) return false;
	if (y < 1 || y > m) return false;
	return true;
}
 
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(Flow - ret, j -> w));
			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);
	nm = n * m;
	for (int i = 1; i <= n; ++i)
		for (int j = 1; j <= m; ++j)
			f[i][j] = (i - 1) * m + j;
	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", &B[i][j]);
			Ans += B[i][j] * 2;
		}
	Tot = 2 * nm; S = ++Tot; T = ++Tot;
	int x, y;
	for (int i = 1; i <= n; ++i)
		for (int j = 1; j <= m; ++j)
		{
			if ((i + j) & 1) 
			{
				AddEdge(f[i][j], T, A[i][j]); 
				AddEdge(S, f[i][j], B[i][j]);
				AddEdge(nm + f[i][j], T, B[i][j]);
				AddEdge(f[i][j], nm + f[i][j], INF);
				for (int k = 0; k < 4; ++k)
				{
					x = i + Dx[k]; y = j + Dy[k];
					if (!Inside(x, y)) continue;
					AddEdge(f[x][y], nm + f[i][j], INF);
				}
			}
			else 
			{
				AddEdge(S, f[i][j], A[i][j]);
				AddEdge(f[i][j], T, B[i][j]);
				AddEdge(S, nm + f[i][j], B[i][j]);
				AddEdge(nm + f[i][j], f[i][j], INF);
				for (int k = 0; k < 4; ++k)
				{
					x = i + Dx[k]; y = j + Dy[k];
					if (!Inside(x, y)) continue;
					AddEdge(nm + f[i][j], f[x][y], INF);
				}	
			}
		}
	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) Ans -= DFS(S, INF);
	printf("%d\n", Ans);
	return 0;
}