Advertisement
Guest User

Untitled

a guest
Oct 15th, 2019
119
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.11 KB | None | 0 0
  1. /*
  2.  
  3. ∧_∧
  4. ( ・ω・。)つ━☆・*。
  5. ⊂  ノ    ・゜
  6. しーJ   Accepted
  7.  
  8. */
  9.  
  10.  
  11.  
  12. // #pragma GCC optimize("O3")
  13. // #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
  14.  
  15. #include <bits/stdc++.h>
  16. #include <ext/pb_ds/assoc_container.hpp>
  17. #include <ext/pb_ds/tree_policy.hpp>
  18.  
  19. #define ll long long
  20. #define all(x) begin(x), end(x)
  21. #define x first
  22. #define y second
  23. #define int long long
  24.  
  25. using namespace std;
  26. using namespace __gnu_pbds;
  27.  
  28. typedef long double ld;
  29. template<typename T>
  30. using kawaii_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
  31.  
  32. const ld PI = atan2(0, -1);
  33.  
  34. void seriy() {
  35. ios::sync_with_stdio(0);
  36. cin.tie(0);
  37. cout.tie(0);
  38. cout << fixed << setprecision(14);
  39. #ifdef _offline
  40. freopen("input.txt", "r", stdin);
  41. freopen("output.txt", "w", stdout);
  42. #endif
  43. }
  44.  
  45. const int MAXN = 1e6 + 10;
  46. const int MAXM = 600;
  47. const int INF = 1e18;
  48. const int BASE = 47;
  49. const int MOD = 1e9 + 7;
  50. const int MAXLOG = 21;
  51.  
  52. vector<vector<pair<int, int>>> g;
  53. vector<int> gas, isgas;
  54.  
  55. void dijkstra(int a, vector<int> &d) {
  56. fill(all(d), INF);
  57. d[a] = 0;
  58. set<pair<int, int>> st;
  59. st.insert({0, a});
  60. while(st.size()) {
  61. int u = st.begin()->y;
  62. st.erase(st.begin());
  63. for(auto v : g[u]) {
  64. if(d[u] + v.y < d[v.x]) {
  65. st.erase({d[v.x], v.x});
  66. d[v.x] = d[u] + v.y;
  67. st.insert({d[v.x], v.x});
  68. }
  69. }
  70. }
  71. }
  72.  
  73. signed main() {
  74. seriy();
  75. int k, n, m, st, en;
  76. cin >> k >> n >> m >> st >> en;
  77. g.resize(n);
  78. isgas.resize(n);
  79. for(int i = 0; i < m; i++) {
  80. int u, v, w;
  81. cin >> u >> v >> w;
  82. u--;
  83. v--;
  84. g[u].push_back({v, w});
  85. g[v].push_back({u, w});
  86. }
  87. int l;
  88. cin >> l;
  89. gas.resize(l);
  90. for(int i = 0; i < l; i++) {
  91. cin >> gas[i];
  92. gas[i]--;
  93. isgas[gas[i]] = 1;
  94. }
  95. vector<int> ds(n), de(n);
  96. dijkstra(st, ds);
  97. dijkstra(en, de);
  98.  
  99. return 0;
  100. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement