题目地址:
https://www.luogu.com.cn/problem/P3387
题目描述:
给定一个
n
n
n个点
m
m
m条边有向图,每个点有一个权值,求一条路径,使路径经过的点权值之和最大。你只需要求出这个权值和。允许多次经过一条边或者一个点,但是,重复经过的点,权值只计算一次。
输入格式:
第一行两个正整数
n
,
m
n,m
n,m
第二行
n
n
n个整数,其中第
i
i
i个数
a
i
a_i
ai表示点
i
i
i的点权。
第三至
m
+
2
m+2
m+2行,每行两个整数
u
,
v
u,v
u,v,表示一条
u
→
v
u→v
u→v的有向边。
输出格式:
共一行,最大的点权之和。
数据范围:
1
≤
n
≤
1
0
4
1≤n≤10^4
1≤n≤104,
1
≤
m
≤
1
0
5
1≤m≤10^5
1≤m≤105,
0
≤
a
i
≤
1
0
3
0≤a_i≤10^3
0≤ai≤103。
可以用Tarjan算法按强连通分量进行缩点,新图中每个点的权值为其缩点前点权值之和,缩点之后的图是拓扑图,接着按拓扑逆序递推即可(当然按照动态规划状态定义不同,是按照拓扑正序还是逆序递需要具体考虑。同时,Tarjan算法缩点完成之后,不需要真的进行拓扑排序,因为点按照编号从小到大已经是拓扑逆序了)。在新图中,设 f [ u ] f[u] f[u]是从 u u u出发走出的路径的最大点权,设 c [ u ] c[u] c[u]是 u u u的点权,则: f [ u ] = c [ u ] + max u → v f [ v ] f[u]=c[u]+\max_{u\to v}f[v] f[u]=c[u]+u→vmaxf[v]代码如下:
#include <iostream>
#include <cstring>
using namespace std;
const int N = 1e4 + 10, M = 2e5 + 10;
int n, m, c[N];
int h[N], hs[N], e[M], ne[M], idx;
int dfn[N], low[N], timestamp;
int stk[N], top, scc_cnt;
bool in_stk[N];
int id[N], sum[N], f[N];
void add(int a, int b, int h[]) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void tarjan(int u) {
dfn[u] = low[u] = ++timestamp;
stk[top++] = u, in_stk[u] = true;
for (int i = h[u]; ~i; i = ne[i]) {
int v = e[i];
if (!dfn[v]) {
tarjan(v);
low[u] = min(low[u], low[v]);
} else if (in_stk[v]) low[u] = min(low[u], dfn[v]);
}
if (dfn[u] == low[u]) {
scc_cnt++;
int y;
do {
y = stk[--top];
in_stk[y] = false;
id[y] = scc_cnt;
// 累加点权
sum[scc_cnt] += c[y];
} while (y != u);
}
}
int main() {
memset(h, -1, sizeof h);
memset(hs, -1, sizeof hs);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) scanf("%d", &c[i]);
for (int i = 1; i <= m; i++) {
int a, b;
scanf("%d%d", &a, &b);
add(a, b, h);
}
for (int i = 1; i <= n; i++)
if (!dfn[i]) tarjan(i);
// 建立新图
for (int i = 1; i <= n; i++)
for (int j = h[i]; ~j; j = ne[j])
if (id[i] != id[e[j]])
add(id[i], id[e[j]], hs);
int res = 0;
// 新图中每个点按编号从小到大的顺序即为拓扑逆序
for (int i = 1; i <= scc_cnt; i++) {
int t = 0;
for (int j = hs[i]; ~j; j = ne[j])
t = max(t, f[e[j]]);
f[i] = sum[i] + t;
res = max(res, f[i]);
}
printf("%d\n", res);
}
时空复杂度 O ( n + m ) O(n+m) O(n+m)。