Advertisement
MinhNGUYEN2k4

CENTRE28 || DIJKSTRA

Jan 17th, 2021
132
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.49 KB | None | 0 0
  1. //Nguyen Huu Hoang Minh
  2. #include <bits/stdc++.h>
  3. #define sz(x) int(x.size())
  4. #define all(x) x.begin(),x.end()
  5. #define reset(x) memset(x, 0,sizeof(x))
  6. #define pb push_back
  7. #define mp make_pair
  8. #define fi first
  9. #define se second
  10. #define N 30005
  11. #define remain(x) if (x > MOD) x -= MOD
  12. #define ii pair<int, int>
  13. #define vi vector<int>
  14. #define vii vector< ii >
  15. #define bit(x, i) (((x) >> (i)) & 1)
  16. #define Task "test"
  17. #define int long long
  18.  
  19. using namespace std;
  20.  
  21. typedef long double ld;
  22.  
  23. int n,m;
  24. vector<ii> a[N];
  25. int ds[N], dt[N];
  26. int fs[N], ft[N];
  27.  
  28. void readfile()
  29. {
  30.     ios_base::sync_with_stdio(false);
  31.     cin.tie(0);cout.tie(0);
  32.     if (fopen(Task".inp","r"))
  33.     {
  34.         freopen(Task".inp","r",stdin);
  35.         //freopen(Task".out","w",stdout);
  36.     }
  37.     cin >> n >> m;
  38.     for(int i=1; i<=m; i++)
  39.     {
  40.         int u,v,z;
  41.         cin >> u >> v >> z;
  42.         a[u].pb(ii(v,z));
  43.         a[v].pb(ii(u,z));
  44.     }
  45. }
  46.  
  47. void dijkstra()
  48. {
  49.     priority_queue<ii, vii, greater<ii>> q;
  50.     for(int i=1; i<=n; i++) ds[i] = dt[i] = 99999999999;
  51.     ds[1] = 0;
  52.     fs[1] = 1;
  53.     q.push(ii(0,1));
  54.     while (q.size())
  55.     {
  56.         int u = q.top().se;
  57.         int du = q.top().fi;
  58.         q.pop();
  59.         if (du != ds[u]) continue;
  60.         for(int i=0; i<a[u].size(); i++)
  61.         {
  62.             int v = a[u][i].fi;
  63.             int uv = a[u][i].se;
  64.             if (ds[v] > ds[u] + uv)
  65.             {
  66.                 ds[v] = ds[u] + uv;
  67.                 fs[v] = fs[u];
  68.                 q.push(ii(ds[v],v));
  69.             }
  70.             else if (ds[v] == ds[u] + uv) fs[v] += fs[u];
  71.         }
  72.     }
  73.     dt[n] = 0;
  74.     ft[n] = 1;
  75.     q.push(ii(0,n));
  76.     while (q.size())
  77.     {
  78.         int u = q.top().se;
  79.         int du = q.top().fi;
  80.         q.pop();
  81.         if (du != dt[u]) continue;
  82.         for(int i=0; i<a[u].size(); i++)
  83.         {
  84.             int v = a[u][i].fi;
  85.             int uv = a[u][i].se;
  86.             if (dt[v] > dt[u] + uv)
  87.             {
  88.                 dt[v] = dt[u] + uv;
  89.                 ft[v] = ft[u];
  90.                 q.push(ii(dt[v],v));
  91.             }
  92.             else if (dt[v] == dt[u] + uv) ft[v] += ft[u];
  93.         }
  94.     }
  95. }
  96.  
  97. void proc()
  98. {
  99.     dijkstra();
  100.     int D = ds[n];
  101.     vector<int> res;
  102.     for(int i=2; i<n; i++)
  103.     {
  104.         if (ds[i] + dt[i] > D || fs[i]*ft[i] < fs[n]) res.pb(i);
  105.     }
  106.     cout << res.size() << '\n';
  107.     for(auto u : res) cout << u << '\n';
  108. }
  109.  
  110. signed main()
  111. {
  112.     readfile();
  113.     proc();
  114.     return 0;
  115. }
  116.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement