Advertisement
Korotkodul

CF D 4

Aug 15th, 2022 (edited)
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 7.76 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 showMN(){
  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.         }cout<<"\n";
  86.     }
  87.  
  88.     void showfr0(){
  89.         cout<<"fr0\n";
  90.         int id=-1;
  91.         for (int lg = 0; lg <= log2(N) - 1; ++lg){
  92.             for (int i = 0; i < pow(2, lg); ++i){
  93.                 id++;
  94.                 seep(fr0[id]);
  95.             }cout<<"\n";
  96.         }cout<<"\n";
  97.     }
  98.  
  99.     void showfr1(){
  100.         cout<<"fr1\n";
  101.         int id = -1;
  102.         for (int lg = 0; lg <= log2(N) - 1; ++lg){
  103.             for (int i = 0; i < pow(2, lg); ++i){
  104.                 id++;
  105.                 seep(fr1[id]);
  106.             }cout<<"\n";
  107.         }cout<<"\n";
  108.     }
  109.  
  110.     void bld(){
  111.         N = up2(n);
  112.         MN.assign(2 * N - 1, {inf,-1});
  113.         for (int i = N - 1; i < N - 1 + n; ++i){
  114.             int id = i - ( N - 1);
  115.             MN[i] = {a[id], id};
  116.         }
  117.         for (int i = N - 2;i >= 0; --i){
  118.             if (MN[2*i + 1] < MN[2 * i + 2]){
  119.                 MN[i] = MN[2*i + 1];
  120.             }
  121.             else{
  122.                 MN[i] = MN[2 * i + 2];
  123.             }
  124.         }
  125.  
  126.         fr0.assign(N - 1, {0,-1});
  127.         fr1.assign(N - 1, {0,-1});
  128.         /*
  129.         УЧТИ?
  130.         fr0
  131.         * * *...
  132.         fr1
  133.         0 * * ...
  134.         здесь первая не учитывается и это надо учесть: как?
  135.  
  136.         вот как:
  137.         fr0
  138.         левый нижний отрезок (0,1)
  139.         fr1
  140.         левый ниждний отрезок (1,2)
  141.         просто это надо учитывать и все
  142.         */
  143.         //fr0
  144.         for (int i = 0; i + 1 < n; i += 2){
  145.             int mn = a[i], id = i;
  146.             if (a[i] > a[i+1]){
  147.                 mn = a[i+1];
  148.                 id++;
  149.             }
  150.             fr0[i/2 + N/2 - 1] = {mn, id};
  151.         }
  152.  
  153.         for (int i = 1; i + 1 < n; i += 2){
  154.             int mn = a[i], id = i;
  155.             if (a[i] > a[i + 1]){
  156.                 mn = a[i + 1];
  157.                 id++;
  158.             }
  159.             fr1[i/2 + N/2 - 1] = {mn, id};
  160.         }
  161.  
  162.         for (int i = N/2 - 2; i >= 0; --i){
  163.  
  164.             /*fr0[i] = max(fr0[i*2 + 1], fr0[i * 2 + 2]);
  165.             fr1[i] = max(fr1[i*2 + 1], fr1[i * 2 + 2]);*/
  166.             pii p = fr0[i*2 + 1];
  167.             if (fr0[i*2 + 2].first > fr0[i*2 + 1].first){
  168.                 p = fr0[i*2 + 2];
  169.             }
  170.             fr0[i] = p;
  171.         }
  172.  
  173.         for (int i = N/2 - 2; i >= 0; --i){
  174.             pii p = fr1[i*2 + 1];
  175.             if (fr1[i*2 + 2].first > fr1[i*2 + 1].first){
  176.                 p = fr1[i*2 + 2];
  177.             }
  178.             fr1[i] = p;
  179.         }
  180.  
  181.  
  182.         if (sh){
  183.             //showMN();
  184.             //showfr0();
  185.         }
  186.     }
  187.  
  188.  
  189.     void upd(int id){
  190.         //!!!
  191.         a[id] = inf;
  192.  
  193.         //if id % 2 == 0
  194.         int id0, id1;
  195.         id0 = (id + N - 1) / 2;
  196.         id1 = (id + N - 1 - 1) / 2;
  197.  
  198.         if (sh){
  199.             cout<<"id0 id1 = "<<id0<<' '<<id1<<"\n";
  200.         }
  201.         int to0, to1; //соседи по паре в fr0 и fr1
  202.         bool upd0=1, upd1=1;
  203.         if (id == 0){
  204.             if (sh){
  205.                 cout<<"A\n";
  206.             }
  207.             upd1 = 0;
  208.         }
  209.         else if (n % 2 == 0 && id == n - 1){
  210.             if (sh){
  211.                 cout<<"B\n";
  212.             }
  213.             upd1 = 0;
  214.         }
  215.  
  216.         if (n % 2 == 1 && id == n - 1){
  217.             if (sh){
  218.                 cout<<"C\n";
  219.             }
  220.             upd0 = 0;
  221.         }
  222.  
  223.         if (id % 2 == 0){
  224.             if (sh){
  225.                 cout<<"D\n";
  226.             }
  227.             to0 = id+1;
  228.             to1 = id-1;
  229.         }
  230.         else if (id % 2 == 1){
  231.             if (sh){
  232.                 cout<<"E\n";
  233.             }
  234.             to0 = id-1;
  235.             to1 = id+1;
  236.         }
  237.  
  238.         //fr0
  239.         if (upd0){
  240.             if (sh){
  241.                 cout<<"F\n";
  242.             }
  243.             fr0[id0] = {a[id], id};
  244.             if (a[to0] < a[id]){
  245.                 fr0[id0] = {a[to0], to0};
  246.             }
  247.         }
  248.  
  249.         while (upd0 && id0 > 0){
  250.             id0--;
  251.             id0 /= 2;
  252.             pii p = fr0[id0 * 2 + 1];
  253.             if (fr0[id0 * 2 + 2].first > fr0[id0 * 2 + 1].first){
  254.                 p = fr0[id0 * 2 + 2];
  255.             }
  256.             fr0[id0] = p;
  257.         }
  258.  
  259.         //fr1
  260.         if (upd1){
  261.             if (sh){
  262.                 cout<<"G\n";
  263.             }
  264.             fr1[id1] = {a[id], id};
  265.             if (a[to1] < a[id]){
  266.                 fr1[id1] = {a[to1], to1};
  267.             }
  268.         }
  269.  
  270.         while (upd1 && id1 > 0){
  271.             id1--;
  272.             id1 /= 2;
  273.             pii p = fr1[id1 * 2 + 1];
  274.             if (fr1[id1 * 2 + 2].first > fr1[id1 * 2 + 1].first){
  275.                 p = fr1[id1 * 2 + 2];
  276.             }
  277.             fr1[id1] = p;
  278.         }
  279.  
  280.         if (sh){
  281.             cout<<"a\n"; cv(a);
  282.             cout<<"upd0 upd1 = "<<upd0<<" "<<upd1<<"\n";
  283.             cout<<"to0 to1 = "<<to0<<" "<<to1<<"\n";
  284.             cout<<"id0 id1 = "<<id0<<" "<<id1<<"\n";
  285.             if (upd0){
  286.                 cout<<"fr0[id0] = "; seep(fr0[id0]);
  287.             }
  288.             if (upd1){
  289.                 cout<<"fr1[id1] = "; seep(fr1[id1]);
  290.             }
  291.             cout<<"\n";
  292.         }
  293.  
  294.         //MN
  295.         id += N - 1;
  296.         MN[id].first = inf;
  297.         while (id > 0){
  298.             id--;
  299.             id /= 2;
  300.             pii p = MN[2 * id + 1];
  301.             if (MN[2 * id + 1].first > MN[2 * id + 2].first){
  302.                 p = MN[2 * id + 2];
  303.             }
  304.             MN[id] = p;
  305.         }
  306.  
  307.  
  308.     }
  309. };
  310.  
  311. void slv(){
  312.     tree T;
  313.     T.bld();
  314.     if (sh){
  315.         cout<<"a\n"; cv(a);
  316.         T.showMN();
  317.         T.showfr0();
  318.         T.showfr1();
  319.     }
  320.     if (sh){
  321.         int q; cin>>q;
  322.         for (int i = 0; i < q; ++i){
  323.             int id; cin>>id;
  324.             T.upd(id);
  325.             if (sh){
  326.                 cout<<"a\n"; cv(a);
  327.                 T.showMN();
  328.                 T.showfr0();
  329.                 T.showfr1();
  330.             }
  331.  
  332.             //T.showfr1();
  333.         }
  334.     }
  335. }
  336.  
  337.  
  338. int main()
  339. {
  340.     /*ios::sync_with_stdio(0);
  341.     cin.tie(0);
  342.     cout.tie(0);*/
  343.     int t=1; if (!sh) cin>>t;
  344.     for (int go = 0; go < t; ++go){
  345.         cin>>n>>k; a.resize(n); for (int &i: a) cin>>i;
  346.         slv();
  347.     }
  348. }
  349. /*
  350. 20 20
  351. 1 4 3 6 3
  352. 1 4 7 4 7
  353. 1 4 3 6 4
  354. 1 4 3 2 3
  355. */
  356.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement