2019 Multi-University Training Contest 3：K Subsequence（dijkstra费用流（模板））

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6611


题目大意：给你一个序列，让你找k个不相交的不下降子序列，使得子序列所有元素的值之和最大。

题解：一道非常裸的题。网络流可以用来求不相交路径，做法是拆点，使得每个点只能用一次，只要按要求建有向图即可。要和最大，就是费用流，由于出题人卡掉了spfa版的费用流，得写dijkstra费用流（BF费用流也可以）。这题可以作为dijkstra版的费用流板子题来练习。

关于dijkstra费用流：就是用dijkstra代替spfa来找增广路径，由于dijkstra只能处理非负边权图，在负边权图需要做处理：引入一个势能函数，每一次松弛：dis[u] > dis[v] + w(u,v) 改为 dis[u] > dis[v] + w(u,v) + h[u] - h[v]。h[u],h[v] 就是势能函数。只要保证w(u,v) + h[u] - h[v] 是正的 ，那么整张图就没有负边权。
h[u],h[v]如何构造：考虑原图最短路松弛：对于某一条边 w(u,v)，有dist[u] >= dist[v] + w(u,v)，dist[u] - dist[v] >= w(u,v)。因此可以让势能函数 h[u] = dist[u]，其中dist[u] 为上一次原图最短路松弛得到的dist函数值（也就是原本传统的dist）。这样松弛时的边权w(u,v) 被换成了 w(u,v) + h[u] - h[v]，dis[u] 存的实际上是$\sum w(u,v)+h[u]-h[v]$ = dist[u] + h[s] - h[u] = dist[u] - h[u] (注意dist[u] 是原图第i次增广的dist数组，和dis不同，dis数组因为引入了势能函数意义已经改变)；

每一次增广h[u] 必须是上一次的增广的dist[u]，所以每次增广完后 h[u] 要 加上 dis[u] (dis[u] = dist[u] - h[u])，这样h[u]就变成了 这次增广的 dist[u]，可以供下一次增广使用。

还一个注意点：dijkstra费用流不能写前向星（在这题会T掉，我也不知道为什么），得写vector。
vector的结构题多了一个rev，rev是用来找反向弧的，网络流中一条边(u,v)（存入了g[u]）， 对应一条反向弧(v,u)（存入了g[v]） ,rev 就是用来存u在 g[v] 的位置，同理 v 也有一个rev 存的是 v 在 g[u]中的位置。这样的好处就是在增广时可以根据preV,preE数组沿汇点回到起点。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e4;
const int maxm = 2e4 + 10;
const int inf = 0x3f3f3f3f;
#define pii pair<int,int>
#define fir first
#define sec second
int n,k,t;
struct node{
	int v,c,w,rev;		//rev 是 反向弧(u,v)，v 在 u 的位置。
	node(int vi = 0,int ci = 0,int wi = 0,int ri = 0) {
		v = vi;c = ci;w = wi;rev = ri;
	}
};
int cnt,head[maxn],dis[maxn],h[maxn],vis[maxn],pre[maxn],a[maxn];
vector<node> g[maxn];
void init() {
	cnt = 0;
	for(int i = 0; i < maxn; i++) g[i].clear();
}
int preV[maxn],preE[maxn];
void add(int u,int v,int c,int w) {
	g[u].push_back(node(v,c,w,(int)g[v].size()));
	g[v].push_back(node(u,0,-w,(int)(g[u].size() - 1)));
}
int maxflow(int s,int t,int flow = inf) {
	int ans = 0,f = 0;
	fill(h,h + t + 1,0);
	fill(preV,preV + t + 1,0);
	fill(preE,preE + t + 1,0);
	while(flow) {
		priority_queue<pii,vector<pii>,greater<pii> > q;
		fill(dis,dis + t + 1,inf);
		dis[s] = 0;q.push(pii(dis[s],s));
		while(!q.empty()) {
			pii now = q.top();
			q.pop();
			int u = now.sec;
			if(dis[u] < now.fir) continue;
			for(int i = 0; i < g[u].size(); i++) {
				int v = g[u][i].v,c = g[u][i].c,w = g[u][i].w;
				if(c && dis[v] > w + dis[u] + h[u] - h[v]) {
					preV[v] = u;preE[v] = i;
					dis[v] = w + dis[u] + h[u] - h[v];
					q.push(pii(dis[v],v));
				}
			}
		}		
		if(dis[t] == inf) break;	
		for(int i = 0; i <= t + 1; i++) h[i] += dis[i];
		int mx = inf;
		for(int i = t; i != s; i = preV[i])
			mx = min(mx,g[preV[i]][preE[i]].c);
		flow -= mx;f += mx;ans += h[t] * mx;
		for(int i = t; i != s; i = preV[i]) {
			g[preV[i]][preE[i]].c -= mx;
			g[i][g[preV[i]][preE[i]].rev].c += mx;
		}
	}
	return -ans;
}
int main() {
	scanf("%d",&t);
	while(t--) {
		init();
		scanf("%d%d",&n,&k);
		for(int i = 1; i <= n; i++)
			scanf("%d",&a[i]);
		int mx = 0;
		int s = 0,t = 2 * n + 1;
		for(int i = 1; i <= n; i++) {
			add(i,i + n,1,-a[i]);
			add(s,i,1,0);
			add(i + n,t,1,0);
			for(int j = 1; j < i; j++)
				if(a[j] <= a[i])
					add(j + n,i,1,0);
		}
		int ans = maxflow(s,t,k);
		printf("%d\n",ans);
	}
	return 0;
}