[luogu_P2045]方格取数加强版

[luogu_P2045]方格取数加强版

试题描述

给出一个 \(n \times n\) 的矩阵,每一格有一个非负整数 \(A_{i,j},(A_{i,j} \le 1000)\) 现在从 \((1,1)\) 出发,可以往右或者往下走,最后到达 \((n,n)\),每达到一格,把该格子的数取出来,该格子的数就变成 \(0\),这样一共走 \(K\) 次,现在要求 \(K\) 次所达到的方格的数的和最大

输入

第一行两个数 \(n,k\)（\(1 \le n \le 50, 0 \le k \le 10\)）

接下来 \(n\) 行,每行 \(n\) 个数,分别表示矩阵的每个格子的数

输出

一个数,为最大和

输入示例

3 1
1 2 3
0 2 1
1 4 2

输出示例

11

数据规模及约定

见“试题描述”和“输入”

题解

网格图拆点后建边，源点到 \((1, 1)\) 容量限制为 \(k\)，跑最小费用最大流。

假设点 \(i\) 拆成 \(in_i\) 和 \(out_i\)，则在 \(in_i\) 和 \(out_i\) 之间连两条边，一条容量为 \(1\)，费用为对应方格中的数的相反数，另一条容量无穷费用为 \(0\)。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cctype>
#include <algorithm>
#include <queue>
using namespace std;
#define rep(i, s, t) for(int i = (s); i <= (t); i++)
#define dwn(i, s, t) for(int i = (s); i >= (t); i--)

int read() {
	int x = 0, f = 1; char c = getchar();
	while(!isdigit(c)){ if(c == '-') f = -1; c = getchar(); }
	while(isdigit(c)){ x = x * 10 + c - '0'; c = getchar(); }
	return x * f;
}

#define maxn 5010
#define maxm 40010
#define oo 2147483647

struct Edge {
	int from, to, flow, cost;
	Edge() {}
	Edge(int _1, int _2, int _3, int _4): from(_1), to(_2), flow(_3), cost(_4) {}
};
struct ZKW {
	int n, m, s, t, cost, ans, head[maxn], nxt[maxm];
	Edge es[maxm];
	int d[maxn];
	deque <int> Q;
	bool inq[maxn];
	bool vis[maxn];

	void init() {
		m = 0; memset(head, -1, sizeof(head));
		return ;
	}
	void setn(int _) {
		n = _;
		return ;
	}

	void AddEdge(int a, int b, int c, int d) {
		es[m] = Edge(a, b, c, d); nxt[m] = head[a]; head[a] = m++;
		es[m] = Edge(b, a, 0, -d); nxt[m] = head[b]; head[b] = m++;
		return ;
	}

	bool BFS() {
		rep(i, 1, n) d[i] = oo;
		d[t] = 0;
		Q.push_back(t); inq[t] = 1;
		while(!Q.empty()) {
			int u = Q.front(); Q.pop_front(); inq[u] = 0;
			for(int i = head[u]; i != -1; i = nxt[i]) {
				Edge& e = es[i^1];
				if(d[e.from] > d[u] + e.cost && e.flow) {
					d[e.from] = d[u] + e.cost;
					if(!inq[e.from]) {
						inq[e.from] = 1;
						if(Q.empty() || d[e.from] <= d[Q.front()]) Q.push_front(e.from);
						else Q.push_back(e.from);
					}
				}
			}
		}
		if(d[s] == oo) return 0;
		cost = d[s];
		return 1;
	}

	int DFS(int u, int a) {
		if(u == t || !a) return ans += cost * a, a;
		if(vis[u]) return 0;
		vis[u] = 1;
		int flow = 0, f;
		for(int i = head[u]; i != -1; i = nxt[i]) {
			Edge& e = es[i];
			if(d[e.to] == d[u] - e.cost && (f = DFS(e.to, min(a, e.flow)))) {
				flow += f; a -= f;
				e.flow -= f; es[i^1].flow += f;
				if(!a) return flow;
			}
		}
		return flow;
	}

	int MaxFlow(int _s, int _t) {
		s = _s; t = _t;
		int flow = 0, f;
		while(BFS())
			do {
				memset(vis, 0, sizeof(vis));
				f = DFS(s, oo);
				flow += f;
			} while(f);
		return flow;
	}
} sol;

#define maxr 55
int CntP;
struct Point {
	int id;
	Point(): id(0) {}
	int p() { return id ? id : id = ++CntP; }
} In[maxr][maxr], Out[maxr][maxr], S, T;

int main() {
	int n = read(), k = read();
	sol.init();
	sol.AddEdge(S.p(), In[1][1].p(), k, 0);
	sol.AddEdge(Out[n][n].p(), T.p(), oo, 0);
	rep(i, 1, n) rep(j, 1, n) {
		sol.AddEdge(In[i][j].p(), Out[i][j].p(), 1, -read());
		sol.AddEdge(In[i][j].p(), Out[i][j].p(), oo, 0);
		if(i < n) sol.AddEdge(Out[i][j].p(), In[i+1][j].p(), oo, 0);
		if(j < n) sol.AddEdge(Out[i][j].p(), In[i][j+1].p(), oo, 0);
	}
	sol.setn(CntP);

	sol.MaxFlow(S.p(), T.p());
	printf("%d\n", -sol.ans);

	return 0;
}