Advertisement
MinhNGUYEN2k4

ttrip

Mar 23rd, 2021
176
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.14 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 105
  11. #define remain(x) if (x > MOD) x -= MOD
  12. #define ii pair<int, int>
  13. #define iiii pair< ii , ii >
  14. #define viiii vector< iiii >
  15. #define vi vector<int>
  16. #define vii vector< ii >
  17. #define bit(x, i) (((x) >> (i)) & 1)
  18. #define Task "test"
  19. #define int long long
  20.  
  21. using namespace std;
  22.  
  23. typedef long double ld;
  24. const int inf = 1e10;
  25. const int minf = -1e10;
  26.  
  27. int n, a[N][N];
  28.  
  29. void readfile()
  30. {
  31.     ios_base::sync_with_stdio(false);
  32.     cin.tie(0);cout.tie(0);
  33.     if (fopen(Task".inp","r"))
  34.     {
  35.         freopen(Task".inp","r",stdin);
  36.         //freopen(Task".out","w",stdout);
  37.     }
  38.     cin >> n;
  39.     for(int i=1; i<=n; i++)
  40.     {
  41.         for(int j=1; j<=n; j++)
  42.         {
  43.             cin >> a[i][j];
  44.             a[i][j] = (a[i][j]==0) ? inf : a[i][j];
  45.         }
  46.     }
  47. }
  48.  
  49. void floyd()
  50. {
  51.     for(int k=1; k<=n; k++)
  52.         for(int i=1; i<=n; i++)
  53.             for(int j=1; j<=n; j++)
  54.                 a[i][j] = min(a[i][j],a[i][k] + a[k][j]);
  55. }
  56.  
  57. void proc()
  58. {
  59.     priority_queue<ii, vii, greater<ii>> g[n+1];
  60.     vector<bool> vst(n+1,false);
  61.     floyd();
  62.     for(int i=1; i<=n; i++)
  63.         for(int j=1; j<n; j++)
  64.             g[i].push(ii(a[i][j],j));
  65.     int res = 0, u = 1;
  66.     while (true)
  67.     {
  68.         vst[u] = true;
  69.         while (g[u].size() && vst[g[u].top().se]) g[u].pop();
  70.         //Nếu đã thăm r thì pop nó ra khỏi heap hiện tại
  71.         if (g[u].empty()){
  72.             res += a[u][n];
  73.             break;
  74.         }
  75.         // nếu heap rỗng tức là đã thăm n-1 đỉnh r -> đến đỉnh n và kết thúc
  76.         else{
  77.         // nếu heap không rỗng tức là ta cần đi đến đỉnh v chưa thăm hiện tại trong heap
  78.             int v = g[u].top().se;
  79.             res += g[u].top().fi;
  80.             g[u].pop();
  81.             u = v;
  82.         }
  83.     }
  84.     cout << res;
  85. }
  86.  
  87. signed main()
  88. {
  89.     readfile();
  90.     proc();
  91.     return 0;
  92. }
  93.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement