mr_dot_convict

triaal.cpp

Nov 10th, 2019
101
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.84 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. using ll = long long;
  5. using pii = pair<int,int>;
  6.  
  7. int main () {
  8. int n, m;
  9. cin >> n >> m;
  10. vector <int> w(n);
  11. for (int i = 0; i < n; ++i) {
  12. cin >> w[i];
  13. }
  14. int u, v;
  15. vector<vector <int>> T(n, vector<int>());
  16. for (int i = 0; i < m; ++i){
  17. cin >> u >> v;
  18. --u, --v;
  19. T[u].push_back(v);
  20. T[v].push_back(u);
  21. }
  22. int s;
  23. cin >> s;
  24. --s;
  25.  
  26. vector <bool> vis(n, false);
  27. vector <int> order;
  28. function <void(int)> dfs = [&] (int u) -> void {
  29. vis[u] = true;
  30. for (int v : T[u]) if (!vis[v]) dfs(v);
  31. order.push_back(u);
  32. };
  33. dfs(s);
  34. vector <ll> dp(n);
  35. for (int i = 0; i < n; ++i)
  36. dp[i] = w[i];
  37.  
  38. for (int u : order) {
  39. for (int v : T[u]) {
  40. dp[v] = max(dp[v], dp[v] + dp[u]);
  41. }
  42. }
  43. cout << dp[s] << '\n';
  44. }
Add Comment
Please, Sign In to add comment