网络流_费用流_ACWING382

题目：
382. K取方格数
题目
提交记录
讨论
题解
视频讲解

在一个N*N的矩形网格中，每个格子里都写着一个非负整数。

可以从左上角到右下角安排K条路线，每一步只能往下或往右，沿途经过的格子中的整数会被取走。

若多条路线重复经过一个格子，只取一次。

求能取得的整数的和最大是多少。

输入格式
第一行包含两个整数N和K。

接下来N行，每行包含N个不超过1000的整数，用来描述整个矩形网格。

输出格式
输出一个整数，表示能取得的最大和。

数据范围
1≤N≤50,
0≤K≤10
输入样例：
3 2
1 2 3
0 2 1
1 4 2
输出样例：
15

解题思路：
首先我们可以将一个点拆分成2个点，一个出点，一个为入点，出点与入点之间连接2条边，一条边的容量为1，花费为c, 零一条边的容量为k - 1, 花费为0， 然后在能够到达的点之间连接一条边，容量为k， 花费为0， 这样我们就建立了一张由初始点为源点，终点为汇点的图，然后他们容量的总和为k，因为有花费的边的容量为1， 所以满足了题目中需要走k条路经的需求，在保证了最大流的前提下我们再去用spfa去跑最长路，把每次得到的结果加起来最终就得到了最后的答案。

注意点：
1.数组范围要开够
2.定义边的序号的时候可以由一点技巧，具体看代码
3.出队一定要记得 q.pop()。！！！
AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;

const int N = 5005, M = 3e4 + 5;
const int inf = 0xcfcfcfcf;

int n, k, s, t, ans, maxflow;
int h[N], e[M], w[M], c[M], ne[M], idx;
int d[N], incf[N], pre[N];
bool vis[N];

inline int num(int i, int j, int k) {
	return (i - 1) * n + j + k * n * n;
}
 
inline void add(int a, int b, int we, int co) {
	e[idx] = b, w[idx] = we, c[idx] = co, ne[idx] = h[a], h[a] = idx ++;
	e[idx] = a, w[idx] = 0, c[idx] = - co, ne[idx] = h[b], h[b] = idx ++;
}

inline bool spfa(void) {
	queue<int> q;
	while(q.size()) q.pop();
	memset(vis, false, sizeof vis);
	memset(d, 0xcf, sizeof d);
	incf[s] = 1 << 30;
	
	d[s] = 0; q.push(s); vis[s] = true;
	
	while(q.size()) {
		int u = q.front(); q.pop(); vis[u] = false;
		
		for(int i = h[u]; i != -1; i = ne[i]) {
			int v = e[i];
			if(w[i]) {
				if(d[v] < d[u] + c[i]) {
					d[v] = d[u] + c[i];
					pre[v] = i;
					incf[v] = min(incf[u], w[i]);
					if(!vis[v]) q.push(v), vis[v] = true;
				}
			}
		}
	}
	
	if(d[t] == 0xcfcfcfcf) return false;
	return true;
}

inline void update(void) {
	int x = t;
	while(x != s) {
		int i = pre[x];
		w[i] -= incf[t];
		w[i ^ 1] += incf[t];
		x = e[i  ^ 1];
	}
	
	maxflow += incf[t];
	ans += d[t] * incf[t];
}

int main(void) {
	scanf("%d%d", &n, &k);
	memset(h, -1, sizeof h);
	s = 1, t = 2 * n * n;
	
	for(int i = 1; i <= n; i ++) {
		for(int j = 1; j <= n; j ++) {
			int cost; scanf("%d", &cost);
			add(num(i, j, 0), num(i, j, 1), 1, cost);
			add(num(i, j, 0), num(i, j, 1), k - 1, 0);
			if(j < n) add(num(i, j, 1), num(i, j + 1, 0), k, 0);
			if(i < n) add(num(i, j, 1), num(i + 1, j, 0), k, 0);
		}
	}
	
	while(spfa()) update();
	
	printf("%d\n", ans);
	
	return 0;
}