【洛谷】P3387 【模板】缩点

题目地址：

https://www.luogu.com.cn/problem/P3387

题目描述：
给定一个$n$个点$m$条边有向图，每个点有一个权值，求一条路径，使路径经过的点权值之和最大。你只需要求出这个权值和。允许多次经过一条边或者一个点，但是，重复经过的点，权值只计算一次。

输入格式：
第一行两个正整数$n,m$
第二行$n$个整数，其中第$i$个数$a_i$表示点$i$的点权。
第三至$m+2$行，每行两个整数$u,v$，表示一条$u\to v$的有向边。

输出格式：
共一行，最大的点权之和。

数据范围：
$1\le n\le 10^4$，$1\le m\le 10^5$，$0\le a_i\le 10^3$。

可以用Tarjan算法按强连通分量进行缩点，新图中每个点的权值为其缩点前点权值之和，缩点之后的图是拓扑图，接着按拓扑逆序递推即可（当然按照动态规划状态定义不同，是按照拓扑正序还是逆序递需要具体考虑。同时，Tarjan算法缩点完成之后，不需要真的进行拓扑排序，因为点按照编号从小到大已经是拓扑逆序了）。在新图中，设$f[u]$是从$u$出发走出的路径的最大点权，设$c[u]$是$u$的点权，则：$$f[u]=c[u]+\underset{u\to v}{\max }f[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)$。