Advertisement
Korotkodul

CF D 2

Aug 15th, 2022 (edited)
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 5.48 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. #define vec vector
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v){
  17.     for (auto x: v) cout<<x<<' ';
  18.     cout<<"\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v){
  22.     for (auto x: v) cout<<x<<' ';
  23.     cout<<"\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v){
  28.     for (auto x: v) cv(x);
  29.     cout<<"\n";
  30. }
  31.  
  32. void cvb(vector <bool> v){
  33.     for (bool x: v) cout<<x<<' ';
  34.     cout<<"\n";
  35. }
  36.  
  37. void cvs(vector <string>  v){
  38.     for (auto a: v){
  39.         cout<<a<<"\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a){
  44.     for (auto p: a){
  45.         cout<<p.first<<' '<<p.second<<"\n";
  46.     }
  47.     cout<<"\n";
  48. }
  49.  
  50. bool sh=1;
  51.  
  52.  
  53. int n,N,k;
  54.  
  55. vector <int> a;
  56.  
  57. const int inf = 1e9;
  58.  
  59. int up2(int x){
  60.     int y = log2(x);
  61.     if (pow(2, y) < x){
  62.         y++;
  63.     }
  64.     y = pow(2, y);
  65.     return y;
  66. }
  67.  
  68. void seep(pii p){
  69.     cout<<"("<<p.first<<","<<p.second<<") ";
  70. }
  71.  
  72. struct tree{
  73.     vector <pii> MN; //{mn, id}
  74.     vector <pii> fr0, fr1; //макс мин на отрезках длины 2 {max mn, id}
  75.  
  76.     void show(){
  77.         cout<<"SHOW\n";
  78.         cout<<"MN\n";
  79.         int id=-1;
  80.         for (int lg = 0; lg <= log2(N); ++lg){
  81.             for (int i = 0; i < pow(2, lg); ++i){
  82.                 id++;
  83.                 seep(MN[id]);
  84.             }cout<<"\n";
  85.         }
  86.     }
  87.  
  88.     void bld(){
  89.         N = up2(n);
  90.         MN.assign(2 * N - 1, {inf,-1});
  91.         for (int i = N - 1; i < N - 1 + n; ++i){
  92.             int id = i - ( N - 1);
  93.             MN[i] = {a[id], id};
  94.         }
  95.         for (int i = N - 2;i >= 0; --i){
  96.             if (MN[2*i + 1] < MN[2 * i + 2]){
  97.                 MN[i] = MN[2*i + 1];
  98.             }
  99.             else{
  100.                 MN[i] = MN[2 * i + 2];
  101.             }
  102.         }
  103.  
  104.         fr0.assign(N - 1, {0,-1});
  105.         fr1.assign(N - 1, {0,-1});
  106.         /*
  107.         УЧТИ?
  108.         fr0
  109.         * * *...
  110.         fr1
  111.         0 * * ...
  112.         здесь первая не учитывается и это надо учесть: как?
  113.  
  114.         вот как:
  115.         fr0
  116.         левый нижний отрезок (0,1)
  117.         fr1
  118.         левый ниждний отрезок (1,2)
  119.         просто это надо учитывать и все
  120.         */
  121.         //fr0
  122.         for (int i = 0; i + 1 < n; i += 2){
  123.             int mn = a[i], id = i;
  124.             if (a[i] > a[i+1]){
  125.                 mn = a[i+1];
  126.                 id++;
  127.             }
  128.             fr0[i/2 + N/2 - 1] = {mn, id};
  129.         }
  130.  
  131.         for (int i = 1; i + 1 < n; i += 2){
  132.             int mn = a[i], id = i;
  133.             if (a[i] > a[i + 1]){
  134.                 mn = a[i + 1];
  135.                 id++;
  136.             }
  137.             fr1[i/2 + N/2 - 1] = {mn, id};
  138.         }
  139.  
  140.         for (int i = N/2 - 2; i >= 0; --i){
  141.  
  142.             /*fr0[i] = max(fr0[i*2 + 1], fr0[i * 2 + 2]);
  143.             fr1[i] = max(fr1[i*2 + 1], fr1[i * 2 + 2]);*/
  144.             pii p = fr0[i*2 + 1];
  145.             if (fr0[i*2 + 2].first > fr0[i*2 + 1].first){
  146.                 p = fr0[i*2 + 2];
  147.             }
  148.             fr0[i] = p;
  149.         }
  150.  
  151.         for (int i = N/2 - 2; i >= 0; --i){
  152.             pii p = fr1[i*2 + 1];
  153.             if (fr1[i*2 + 2].first > fr1[i*2 + 1].first){
  154.                 p = fr1[i*2 + 2];
  155.             }
  156.             fr1[i] = p;
  157.         }
  158.  
  159.  
  160.         if (sh){
  161.             show();
  162.         }
  163.     }
  164.  
  165.  
  166.     void upd(int id){
  167.         //!!!
  168.         a[id] = inf;
  169.  
  170.         //if id % 2 == 0
  171.         int id0, id1;
  172.         id0 = id / 2;
  173.         id1 = (id - 1) / 2;
  174.         int to0, to1; //соседи по паре в fro и fr1
  175.         bool upd0=1, upd1=1;
  176.         if (id == 0){
  177.             upd1 = 0;
  178.         }
  179.         else if (n % 2 == 0 && id == n - 1){
  180.             upd1 = 0;
  181.         }
  182.  
  183.         if (n % 2 == 1 && id == n - 1){
  184.             upd0 = 0;
  185.         }
  186.  
  187.         if (id % 2 == 0){
  188.             to0 = id+1;
  189.             to1 = id-1;
  190.         }
  191.         else if (id % 2 == 1){
  192.             to0 = id-1;
  193.             to1 = id+1;
  194.         }
  195.  
  196.         //fr0
  197.         if (upd0){
  198.             fr0[id0] = {a[id], id};
  199.             if (a[to0] < a[id]){
  200.                 fr0[id0] = {a[to0], to0};
  201.             }
  202.         }
  203.  
  204.         while (upd0 && id0 > 0){
  205.             id0--;
  206.             id0 /= 2;
  207.             pii p = fr0[id * 2 + 1];
  208.             if (fr0[id * 2 + 2].first > fr0[id * 2 + 1].first){
  209.                 p = fr0[id * 2 + 2];
  210.             }
  211.             fr0[id] = p;
  212.         }
  213.  
  214.         //fr1
  215.        
  216.  
  217.         //MN
  218.         id += N - 1;
  219.         MN[id].first = inf;
  220.         while (id > 0){
  221.             id--;
  222.             id /= 2;
  223.             pii p = MN[2 * id + 1];
  224.             if (MN[2 * id + 1].first > MN[2 * id + 2].first){
  225.                 p = MN[2 * id + 2];
  226.             }
  227.             MN[id] = p;
  228.         }
  229.  
  230.  
  231.     }
  232. };
  233.  
  234. void slv(){
  235.     tree T;
  236.     T.bld();
  237.     /*if (sh){
  238.         int q; cin>>q;
  239.         for (int i = 0; i < q; ++i){
  240.             int id; cin>>id;
  241.             T.upd(id);
  242.             T.show();
  243.         }
  244.     }*/
  245. }
  246.  
  247.  
  248. int main()
  249. {
  250.     /*ios::sync_with_stdio(0);
  251.     cin.tie(0);
  252.     cout.tie(0);*/
  253.     int t=1; if (!sh) cin>>t;
  254.     for (int go = 0; go < t; ++go){
  255.         cin>>n>>k; a.resize(n); for (int &i: a) cin>>i;
  256.         slv();
  257.     }
  258. }
  259. /*
  260. 20 20
  261. 1 4 3 6 3
  262. 1 4 7 4 7
  263. 1 4 3 6 4
  264. 1 4 3 2 3
  265. */
  266.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement