Korotkodul

СПБГУ_A_1

Dec 25th, 2021 (edited)
90
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.19 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int,int>
  11. using namespace std;
  12. /*
  13. решение:
  14. 1)сделать граф
  15. 2)найти мин гамильтонов путь
  16. 3) ответ = мин гамильтонов путь + расстояние, чтобы дойти до стартовой вершины этого пути
  17.  
  18.  
  19. ЕСЛИ ТАКОЕ НЕ ЗАЙДЁТ хотя бы на 90 баллов:
  20. )посчитать за О(n^2), из какой вершины выгоднее стартовать = вершина start
  21. )посчитать минимальные гамильтоновы пути из каждой вершины
  22. )перебрать все сочетания за O(n) : (x, y), где x - длина пути до стартовой вершины, y - длина мин гамильтонова пути
  23.  
  24. */
  25. using ll = long long;
  26. using ld = long double;
  27. void cv(vector <int> &v){
  28.     for (auto x: v) cout<<x<<' ';
  29.     cout<<"\n\n";
  30. }
  31. int N;
  32. int inf = 1e9;
  33. vector <int> a, b;
  34. vector <vector <int>> G;
  35. void cG(){
  36.     for (int i = 0; i < N; ++i){
  37.         for (int j = 0; j < N; ++j){
  38.             cout<<"("<<i+1<<","<<j+1<<") = "<<G[i][j]<<"\n";
  39.         }cout<<"\n";
  40.     }cout<<"\n";
  41. }
  42. int mk(int x, int y){
  43.     int idx, idy, r = 0;
  44.     for (int i=0;i<N;++i){
  45.         if (a[i] == x){
  46.             idx = i;
  47.         }
  48.         else if (a[i] == y){
  49.             idy = i;
  50.         }
  51.     }
  52.     r += abs(idx - idy);
  53.  
  54.     for (int i=0;i<N;++i){
  55.         if (b[i] == x){
  56.             idx = i;
  57.         }
  58.         else if (b[i] == y){
  59.             idy = i;
  60.         }
  61.     }
  62.     r += abs(idx - idy);
  63.     return r;
  64. }
  65. vector <vector <int> > dp;
  66. string two(int x){
  67.     string r="";
  68.     while (x > 0){
  69.         r += ((x % 2) + 48);
  70.         x /= 2;
  71.     }
  72.  
  73.     while (r.size() != N){
  74.         r += '0';
  75.     }reverse(r.begin(), r.end());
  76.     return r;
  77. }
  78.  
  79. void wrk(){
  80.     for (int msk = 1; msk < 1 << N; ++msk){
  81.         cout<<"mask = "<<two(msk)<<"\n";
  82.         for (int i = 0; i < N; ++i){
  83.             if (msk & (1 << i) == 0) continue;
  84.             for (int j = 0; j < N; ++j){
  85.                 if (msk && (1 << j) != 0) continue;
  86.                 dp[msk | (1 << j)][j] = min(dp[msk | (1 << j)][j], dp[msk][i] + G[i][j]);
  87.             }
  88.         }
  89.     }
  90. }
  91.  
  92. void cdp(){
  93.     for (auto x: dp){
  94.         for (auto y: x) cout<<y<<' ';
  95.         cout<<"\n";
  96.     }cout<<"\n";
  97. }
  98.  
  99. int main()
  100. {
  101.     ios::sync_with_stdio(0);
  102.     cin.tie(0);
  103.     cout.tie(0);
  104.     cin>>N;
  105.     G.resize(N, vector <int> (N, inf));
  106.     a.resize(N, inf); b.resize(N, inf);
  107.     for (int i = 0; i < N; ++i){
  108.         int inb; cin>>inb; inb--;
  109.         cout<<"inb = "<<inb<<"\n";
  110.         a[i] = i;
  111.         b[inb] = i;
  112.     }
  113.     cv(a);
  114.     cv(b);
  115.     for (int i = 0; i < N; ++i){
  116.         for (int j = 0; j != i && j < N; ++j){
  117.             G[i][j] = G[j][i] = mk(i, j);
  118.         }
  119.     }
  120.     cG();
  121.     //мин гамильтонов путь
  122.     dp.assign(1<<N, vector <int> (N, inf));
  123.     for (int i = 0; i < N; ++i){
  124.         dp[1<<i][i] = 0;
  125.     }
  126.     wrk();
  127. }
  128. /*
  129. 5
  130. 1 3 5 2 4
  131.  
  132.  
  133. 6
  134. 3 6 4 1 5 2
  135. */
  136.  
  137.  
  138.  
Add Comment
Please, Sign In to add comment