【BZOJ-4261】建设游乐场最大费用最大流

4261: 建设游乐场

Time Limit: 50 Sec Memory Limit: 256 MB
Submit: 21 Solved: 8
[Submit][Status][Discuss]

Description

现在有一大块土地，可以看成N*M的方格。在这块土地上，有些格子内是崎岖的山地，无法建造任何东西；其他格子都是平原。现在打算在这块土地上建设一个游乐园。游乐园由若干条闭合的过山车轨道组成，每个平原格子都要铺一截轨道，为下列 6 种类型中的一种：

（每张图表示一块平原格子，图内网格线为辅助线，无实际意义。）

其中前 2 种为直轨道，后 4 种为弯轨道。显然对游客来说，弯轨道更加刺激。

由于每块格子风景各不相同，经过一番研究，现给了N*M个方格中的每个格子一个评估值，意义为：如果该格子修建弯轨道，会给游客们带来多少的愉悦值。现需要一名设计师，帮他设计一种最优的轨道建设方案，使所有格子给游客们带来的愉悦值之和尽量大。（如果没有合法方案，输出 -1）

Input

第一行两个正整数 n, m。

接下来 n 行，每行 m 个数，描述了整块土地。其中 1 表示山地，0 表示平原。接下来 n 行，每行 m 个非负整数，第 i 行第 j 个为 Vi,j，表示格子 (i,j) 修建弯轨道能给游客们带来的愉悦值。

Output

一行一个数，表示最优设计方案中给游客们带来的愉悦值之和。

Sample Input

3 3
1 1 1
1 0 0
1 0 0
48 94 1
78 78 81
1 12 60

Sample Output

231

HINT

N<=150,M<=30,Vi,j<=100

Source

Solution

蛮好的一道题，自己想了一段时间还是能想出来的QwQ

首先黑白染色，然后每个节点拆出横竖两个方向的节点$id_{1}$和$id_{2}$.

连边$S$到黑点，白点连$T$，容量为$2$，费用为$0$，限制两个方向；

黑点向黑点的每个方向连两条边，白点的每个方向向白点连两条边，容量都是$1$，费用一条为$val$，一条为$0$，表示选择转弯或者直道的价值。

然后黑点的每个方向向相邻白点的每个方向连边，容量为$1$，费用为$0$。

对于一个拐弯格子，会获得$2*val$的价值，对于一个直道格子，会获得$1*val$的价值，所以最后的答案就是$MaxCost-sum$。

代码还是很好写的，就是一开始细节出了问题WA了两次QwQ

Code

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<cmath>

#include<queue>

using namespace std;

inline int read()

{

	int x=0,f=1; char ch=getchar();

	while (ch<'0' || ch>'9') {if (ch=='-') f=-1; ch=getchar();}

	while (ch>='0' && ch<='9')  {x=x*10+ch-'0'; ch=getchar();}

	return x*f;

}

#define MAXN 30010

#define INF 0x7fffffff

int N,M,mp[200][50],val[200][50],sum,tot;

struct EdgeNode{

	int next,to,cap,cost;

}edge[500010];

int head[MAXN],cnt=1;

inline void AddEdge(int u,int v,int w,int c) {cnt++; edge[cnt].next=head[u]; head[u]=cnt; edge[cnt].to=v; edge[cnt].cap=w; edge[cnt].cost=c;}

inline void InsertEdge(int u,int v,int w,int c) {AddEdge(u,v,w,c); AddEdge(v,u,0,-c);}

int Cost,S,T,dis[MAXN],mark[MAXN];

queue<int>q;

inline bool spfa()

{

    for (int i=S; i<=T; i++) dis[i]=-INF,mark[i]=0;

    queue<int>q;

    q.push(S); dis[S]=0; mark[S]=1;

    while (!q.empty()) {

        int now=q.front(); q.pop(); mark[now]=0;

        for (int i=head[now]; i; i=edge[i].next)

            if (edge[i].cap && dis[edge[i].to]<dis[now]+edge[i].cost) {

                    dis[edge[i].to]=dis[now]+edge[i].cost;

                    if (!mark[edge[i].to]) q.push(edge[i].to),mark[edge[i].to]=1;

            }

    }

    return dis[T]!=-INF;

}

inline int dfs(int loc,int low)

{

    mark[loc]=1;

    if (loc==T) return low;

    int used=0,w;

    for (int i=head[loc]; i; i=edge[i].next)

        if (edge[i].cap && !mark[edge[i].to] && dis[edge[i].to]==dis[loc]+edge[i].cost) {

            w=dfs(edge[i].to,min(edge[i].cap,low-used));

            edge[i].cap-=w; edge[i^1].cap+=w; used+=w; Cost+=w*edge[i].cost;

            if (low==used) return used;

        }

    return used;

}

inline int zkw()

{

    int re=0;

    while (spfa()) {

        mark[T]=1;

        while (mark[T]) {

            for (int i=S; i<=T; i++) mark[i]=0;

            re+=dfs(S,INF);

        }

    }

    return re;

}

int col[200][50],id[200][50][3],ID,dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};

//id 0 主点 id 1 竖方向 id 3 横方向

inline bool Check(int x,int y) {return x>=1&&x<=N&&y>=1&&y<=M;}

int main()

{

	N=read(),M=read();

	for (int i=1; i<=N; i++)

		for (int j=1; j<=M; j++)

			mp[i][j]=read();

	for (int i=1; i<=N; i++)

		for (int j=1; j<=M; j++)

			val[i][j]=read(),sum+=mp[i][j]? 0:val[i][j],col[i][j]=(i+j)&1;

	for (int i=1; i<=N; i++)

		for (int j=1; j<=M; j++)

			for (int k=0; k<=2; k++) id[i][j][k]=++ID;

	S=0; T=++ID;

	for (int i=1; i<=N; i++)

		for (int j=1; j<=M; j++)

			if (!mp[i][j]) {

				if (col[i][j]) {

					InsertEdge(S,id[i][j][0],2,0);

					InsertEdge(id[i][j][0],id[i][j][1],1,val[i][j]);

					InsertEdge(id[i][j][0],id[i][j][1],1,0);

					InsertEdge(id[i][j][0],id[i][j][2],1,val[i][j]);

					InsertEdge(id[i][j][0],id[i][j][2],1,0);

				} else {

					tot+=2;

					InsertEdge(id[i][j][0],T,2,0);

					InsertEdge(id[i][j][1],id[i][j][0],1,val[i][j]);

					InsertEdge(id[i][j][1],id[i][j][0],1,0);

					InsertEdge(id[i][j][2],id[i][j][0],1,val[i][j]);

					InsertEdge(id[i][j][2],id[i][j][0],1,0);

				}

			}

	for (int i=1; i<=N; i++)

		for (int j=1; j<=M; j++)

			if (!mp[i][j] && col[i][j])

				for (int k=0; k<=3; k++) {

					int tx=i+dx[k],ty=j+dy[k];

					if (Check(tx,ty) && !mp[tx][ty])

						if (k==0 || k==2)

							InsertEdge(id[i][j][1],id[tx][ty][1],1,0);

						else

							InsertEdge(id[i][j][2],id[tx][ty][2],1,0);

				} 

	int flow=zkw();

	if (flow<tot) return puts("-1"),0;

	printf("%d\n",Cost-sum); 

	return 0;

}