P3627 [APIO2009] 抢掠计划 题解

题目

题目说了，我们可以重复经过一些 ATM 机，但是里面的前只能取一次。

思路

如果我们发现有些 ATM 机是连接在一起的，就可以把这一对 ATM 机当做一个 ATM 机处理，而这个 ATM 机的钱就是之前联通 ATM 机钱数的和。将所有 ATM 机合并在一起得到一个 DAG（有向无环图），这就是“缩点”技术。

实现方法

先用 Tarjan 算法求出强连通分量，建一个新图，像上述操作合并节点为一个 DAG，跑一边最长路就行了。

我不推荐用拓补排序求最长路，因为起点的入度可能不为零，处理要比较麻烦一些。我用了 SPFA，求单源最长路，对于所有的终点更新答案，输出，走人。

AC Code：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;
const int N = 500100;
int n, m, s, p;
int a[N], w[N];
struct edge{
	int u, v, nxt;
};
edge ed[N];
int head[N], cnt;
void addedge(int u, int v) {
	cnt++;
	ed[cnt].u = u;
	ed[cnt].v = v;
	ed[cnt].nxt = head[u];
	head[u] = cnt;
}
int dfn[N], low[N], dfncnt, st[N], in_stack[N], tp;
int scc[N], sc, minn[N], tot[N];
int sz[N];
int is[N];
void tarjan(int u) {
	low[u] = dfn[u] = ++dfncnt, st[++tp] = u, in_stack[u] = 1;
	for (int i = head[u]; i; i = ed[i].nxt) {
		int v = ed[i].v;
		if (!dfn[v]) {
			tarjan(v);
			low[u] = min(low[u], low[v]);
		} else if (in_stack[v]) {
			low[u] = min(low[u], dfn[v]);
		}
	}
	if (dfn[u] == low[u]) {
		++sc;
		while (st[tp] != u) {
			tot[sc] += w[st[tp]];
			scc[st[tp]] = sc;
			sz[sc]++;
			in_stack[st[tp]] = 0;
			--tp;
		}
		tot[sc] += w[st[tp]];
		scc[st[tp]] = sc;
		sz[sc]++;
		in_stack[st[tp]] = 0;
		--tp;
	}
}
queue<int> q;
int deg[N], head1[N], cnt1;
edge ed1[N];
void addedge1(int u, int v) {
	cnt1++;
	ed1[cnt1].u = u;
	ed1[cnt1].v = v;
	ed1[cnt1].nxt = head1[u];
	head1[u] = cnt1;
}
int dis[N];
bool vis[N];
void spfa() {
	q.push(scc[s]);
	while (!q.empty()) {
		int now = q.front();
		q.pop();
		vis[now] = 0;
		for (int i = head1[now]; i; i = ed1[i].nxt) {
			int v = ed1[i].v;
			dis[v] = max(dis[v], dis[now] + tot[v]);
			if (!vis[v]) {
				q.push(v);
				vis[v] = 1;
			}
		}
	}
}
int main() {
//	freopen("P3627_1.in", "r", stdin);
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin >> n >> m;
	for (int i = 1; i <= m; i ++) {
		int a, b;
		cin >> a >> b;
		addedge(a, b);
	}
	for (int i = 1; i <= n; i ++) {
		cin >> w[i];
	}
	cin >> s >> p;
	for (int i = 1; i <= p; i ++) {
		cin >> a[i];
	}
	tarjan(s);
	for (int i = 1; i <= m; i ++) {
		if (scc[ed[i].u] != scc[ed[i].v]) {
			addedge1(scc[ed[i].u], scc[ed[i].v]);
			deg[scc[ed[i].v]] ++;
		}
	}
	dis[scc[s]] = tot[scc[s]];
	spfa();
	int ans1 = 0;
	for (int i = 1; i <= p; i ++) {
		ans1 = max(ans1, dis[scc[a[i]]]);
	}
	cout << ans1;
	return 0;
}