-
Notifications
You must be signed in to change notification settings - Fork 18
/
1686.cpp
133 lines (118 loc) · 3.88 KB
/
1686.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
#include <bits/stdc++.h>
using namespace std;
#define print_op(...) ostream& operator<<(ostream& out, const __VA_ARGS__& u)
template<typename A, typename B> print_op(pair<A, B>) { return out << "(" << u.first << ", " << u.second << ")"; }
template<typename T_container, typename T = typename enable_if<!is_same<T_container, string>::value, typename T_container::value_type>::type> print_op(T_container) { out << "{"; string sep; for (const T &x : u) out << sep << x, sep = ", "; return out << "}"; }
template<typename T> void dbg_out(string s, T x) {cerr << "\033[1;35m" << s << "\033[0;32m = \033[33m" << x << "\033[0m\n";}
template<typename T, typename... Args> void dbg_out(string s, T x, Args... args) {for (int i=0, b=0; i<(int)s.size(); i++) if (s[i] == '(' || s[i] == '{') b++; else
if (s[i] == ')' || s[i] == '}') b--; else if (s[i] == ',' && b == 0) {cerr << "\033[1;35m" << s.substr(0, i) << "\033[0;32m = \033[33m" << x << "\033[31m | "; dbg_out(s.substr(s.find_first_not_of(' ', i + 1)), args...); break;}}
#ifdef LOCAL
#define dbg(...) dbg_out(#__VA_ARGS__, __VA_ARGS__)
#else
#define dbg(...)
#endif
#define ar array
#define ll long long
#define ld long double
#define sz(x) ((int)x.size())
#define rep(i, a, b) for (int i = (int)(a); i < (int)(b); i++)
#define all(a) (a).begin(), (a).end()
const int MAX_N = 1e5 + 5;
const int MAX_L = 20;
const int MAX_C = 26;
const ll MOD = 1e9 + 7;
const ll INF = 1e9;
const ld EPS = 1e-9;
struct SCC {
int n, ti;
vector<int> num, id, stk, val;
vector<ll> nval;
vector<vector<int>> adj, dag, comp;
SCC(int _n) : n(_n), ti(0), num(n), id(n, -1), adj(n), val(n) {}
void addEdge(int u, int v) {
adj[u].push_back(v);
}
void addVal(int u, int x) {
val[u] = x;
}
void init() {
for (int u = 0; u < n; u++)
if (!num[u])
dfs(u);
dag.resize(sz(comp));
nval.resize(sz(comp));
for (auto &c : comp)
for (int u : c) {
nval[id[u]] += val[u];
for (int v : adj[u])
if (id[u] != id[v])
dag[id[u]].push_back(id[v]);
}
for (auto &v : dag) {
sort(v.begin(), v.end());
v.erase(unique(v.begin(), v.end()), v.end());
}
}
int dfs(int u) {
int low = num[u] = ++ti;
stk.push_back(u);
for (int v : adj[u]) {
if (!num[v])
low = min(low, dfs(v));
else if (id[v] == -1)
low = min(low, num[v]);
}
if (low == num[u]) {
comp.emplace_back();
do {
id[stk.back()] = (int) sz(comp) - 1;
comp.back().push_back(stk.back());
stk.pop_back();
} while (comp.back().back() != u);
}
return low;
}
ll query() {
vector<bool> vis(sz(comp));
vector<ll> dp(sz(comp));
function<void(int)> dfsDP = [&](int u) {
dp[u] = nval[u];
vis[u] = true;
ll mx = 0;
for (int v : dag[u]) {
if (!vis[v]) dfsDP(v);
mx = max(mx, dp[v]);
}
dp[u] += mx;
};
for (int i = 0; i < sz(comp); i++) {
if (!vis[i]) {
dfsDP(i);
}
}
return *max_element(all(dp));
}
};
void solve(int tc = 0) {
int n, m; cin >> n >> m;
SCC scc(n);
for (int i = 0; i < n; i++) {
int x; cin >> x;
scc.addVal(i, x);
}
for (int i = 0; i < m; i++) {
int u, v; cin >> u >> v; u--; v--;
scc.addEdge(u, v);
}
scc.init();
cout << scc.query() << "\n";
}
signed main() {
ios_base::sync_with_stdio(false); cin.tie(NULL);
int tc = 1;
// cin >> tc;
for (int t = 1; t <= tc; t++) {
// cout << "Case #" << t << ": ";
solve(t);
}
}