缩点（模板题）

题意

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

思路

缩点后做DAG上dp，注意缩点后已经成拓扑序了，因此不用再跑一遍拓扑排序。

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
#include <unordered_set>

using namespace std;

const int N = 10010, M = 100010;

int n, m;
int h[N], hs[N], e[M], ne[M], idx;
int val[N];
int low[N], dfn[N], timestamp;
stack<int> stk;
bool in_stk[N];
int id[N], d[N], scc_cnt;
int dist[N];

void add(int h[],int a,int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void tarjan(int u)
{
    low[u] = dfn[u] = ++ timestamp;
    stk.push(u), in_stk[u] = true;
    for(int i=h[u];~i;i=ne[i]){
        int j = e[i];
        if(!dfn[j]){
            tarjan(j);
            low[u] = min(low[u], low[j]);
        }
        else if(in_stk[j]) low[u] = min(low[u], dfn[j]);
    }
    if(low[u]==dfn[u]){
        int v;
        scc_cnt ++;
        do{
            v = stk.top();
            stk.pop();
            in_stk[v] = false;
            id[v] = scc_cnt;
            d[scc_cnt] += val[v];
        }while(v!=u);
    }
}

int main()
{
    cin >> n >> m;
    memset(h,-1,sizeof(h));
    memset(hs,-1,sizeof(hs));
    for(int i=1;i<=n;i++) cin >> val[i];
    while(m--){
        int a,b;
        cin >> a >> b;
        add(h,a,b);
    }
    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]){
            int k = e[j];
            int a = id[i], b = id[k];
            if(a!=b) add(hs,a,b);
        }
    
    for(int i=scc_cnt;i;i--){
        dist[i] += d[i];
        for(int j=hs[i];~j;j=ne[j]){
            int k = e[j];
            dist[k] = max(dist[k], dist[i]);
        }
    }
    int ans = 0;
    for(int i=1;i<=scc_cnt;i++) ans = max(ans, dist[i]);
    cout << ans << endl;
    return 0;
}