Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- /*
- Notes:
- A little weird of a question cause it may not be obvious to me that topo sort works :D
- */
- #pragma GCC optimize ("O3")
- #pragma GCC target("sse4")
- #include <bits/stdc++.h>
- #define f first
- #define s second
- #define pb push_back
- #define mp make_pair
- #define ts to_string
- #define ub upper_bound
- #define lb lower_bound
- const char nl = '\n';
- using namespace std;
- using ll = long long;
- using vi = vector<int>;
- using vl = vector<ll>;
- using pii = pair<int, int>;
- using pll = pair<ll, ll>;
- using str = string;
- //The following debugging code was taken from Benjamin Qi's template :)
- str ts(char c) { return str(1,c); }
- str ts(bool b) { return b ? "true" : "false"; }
- str ts(const char* s) { return (str)s; }
- str ts(str s) { return s; }
- template<class A> str ts(complex<A> c) {
- stringstream ss; ss << c; return ss.str(); }
- str ts(vector<bool> v) {
- str res = "{"; for(int i = 0;i < (int)v.size(); i++) res += char('0'+v[i]);
- res += "}"; return res; }
- template<size_t SZ> str ts(bitset<SZ> b) {
- str res = ""; for(int i = 0; i < b.size(); i++) res += char('0'+b[i]);
- return res; }
- template<class A, class B> str ts(pair<A,B> p);
- template<class T> str ts(T v) {
- bool fst = 1; str res = "{";
- for (const auto& x: v) {
- if (!fst) res += ", ";
- fst = 0; res += ts(x);
- }
- res += "}"; return res;
- }
- template<class A, class B> str ts(pair<A,B> p) {
- return "("+ts(p.f)+", "+ts(p.s)+")"; }
- void DBG() { cerr << "]" << endl; }
- template<class H, class... T> void DBG(H h, T... t) {
- cerr << ts(h); if (sizeof...(t)) cerr << ", ";
- DBG(t...); }
- #ifdef LOCAL
- #define dbg(...) cerr << "LINE(" << __LINE__ << ") -> [" << #__VA_ARGS__ << "]: [", DBG(__VA_ARGS__)
- #else
- #define dbg(...) 0
- //freopen(".in", "r", stdin);
- //freopen(".out", "w", stdout);
- #endif
- //CODE STARTS HERE-----------------------------------------------------------------------------------------//
- const int N = 1e5 + 10;
- int n, m, c, ans[N], vis[N], in[N];
- int a[N]; vector<pii> adj[N];
- queue<int> bfs;
- signed main(){
- cin.tie(nullptr);
- ios_base::sync_with_stdio(false);
- ifstream cin("timeline.in");
- ofstream cout("timeline.out");
- cin >> n >> m >> c;
- for(int i = 0; i < n; i++){
- cin >> a[i];
- }
- for(int i = 0; i < c; i++){
- int x, y, u;
- cin >> x >> y >> u;
- //use a topological sort since you wanna process nodes with fewer edges first
- x--,y--;
- adj[x].pb({y,u});
- in[y]++;
- }
- for(int i =0 ; i < n; i++)
- if(!in[i])
- bfs.push(i);
- while(bfs.size()){
- int x= bfs.front();
- bfs.pop();
- for(auto&y : adj[x]){
- a[y.f] = max(a[y.f], a[x] + y.s);
- if(--in[y.f] == 0)
- bfs.push(y.f);
- }
- }
- for(int i =0 ; i<n; i++){
- cout << a[i] << "\n";
- }
- }
Add Comment
Please, Sign In to add comment