Advertisement
Korotkodul

CF D 5

Aug 15th, 2022
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 8.01 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.     void updMN(int id){
  189.         //MN
  190.         id += N - 1;
  191.         MN[id].first = inf;
  192.         while (id > 0){
  193.             id--;
  194.             id /= 2;
  195.             pii p = MN[2 * id + 1];
  196.             if (MN[2 * id + 1].first > MN[2 * id + 2].first){
  197.                 p = MN[2 * id + 2];
  198.             }
  199.             MN[id] = p;
  200.         }
  201.     }
  202.  
  203.  
  204.     void updfr0(int id){
  205.         //!!!
  206.         a[id] = inf;
  207.  
  208.         //if id % 2 == 0
  209.  
  210.  
  211.         if (sh){
  212.             cout<<"id = "<<id<<"\n";
  213.         }
  214.         //соседи по паре в fr0 и fr1
  215.  
  216.  
  217.         if (n % 2 == 1 && id == n - 1){
  218.             if (sh){
  219.                 cout<<"C\n";
  220.             }
  221.             return;
  222.         }
  223.  
  224.  
  225.  
  226.         int to;
  227.  
  228.         if (id % 2 == 0){
  229.             if (sh){
  230.                 cout<<"D\n";
  231.             }
  232.             to = id+1;
  233.         }
  234.         else if (id % 2 == 1){
  235.             if (sh){
  236.                 cout<<"E\n";
  237.             }
  238.             to = id-1;
  239.         }
  240.  
  241.  
  242.  
  243.         //fr0
  244.         if (sh){
  245.             cout<<"F\n";
  246.         }
  247.         fr0[(id + N - 1) / 2] = {a[id], id};
  248.  
  249.         if (a[to] < a[id]){
  250.             fr0[(id + N - 1) / 2] = {a[to], to};
  251.         }
  252.  
  253.         id = (id + N - 1) / 2;
  254.  
  255.         while (id > 0){
  256.             id--;
  257.             id /= 2;
  258.             pii p = fr0[id * 2 + 1];
  259.             if (fr0[id * 2 + 2].first > fr0[id * 2 + 1].first){
  260.                 p = fr0[id * 2 + 2];
  261.             }
  262.             fr0[id] = p;
  263.         }
  264.  
  265.  
  266.  
  267.         /*if (sh){
  268.             cout<<"a\n"; cv(a);
  269.             cout<<"upd0 upd1 = "<<upd0<<" "<<upd1<<"\n";
  270.             cout<<"to0 to1 = "<<to0<<" "<<to1<<"\n";
  271.             cout<<"id0 id1 = "<<id0<<" "<<id1<<"\n";
  272.             if (upd0){
  273.                 cout<<"fr0[id0] = "; seep(fr0[id0]);
  274.             }
  275.             if (upd1){
  276.                 cout<<"fr1[id1] = "; seep(fr1[id1]);
  277.             }
  278.             cout<<"\n";
  279.         }*/
  280.  
  281.  
  282.  
  283.  
  284.     }
  285.  
  286.     void updfr1(int id){
  287.         if (id == 0){
  288.             if (sh){
  289.                 cout<<"A\n";
  290.             }
  291.             return;
  292.         }
  293.         else if (n % 2 == 0 && id == n - 1){
  294.             if (sh){
  295.                 cout<<"B\n";
  296.             }
  297.             return;
  298.         }
  299.  
  300.         int to;
  301.         if (id % 2 == 0){
  302.             if (sh){
  303.                 cout<<"D\n";
  304.             }
  305.             to = id-1;
  306.         }
  307.         else if (id % 2 == 1){
  308.             if (sh){
  309.                 cout<<"E\n";
  310.             }
  311.             to = id+1;
  312.         }
  313.         id--;
  314.  
  315.         //fr1
  316.         if (sh){
  317.             cout<<"G\n";
  318.         }
  319.         fr1[(id + N - 1) / 2] = {a[id], id};
  320.         if (a[to] < a[id]){
  321.             fr1[(id + N - 1) / 2] = {a[to], to};
  322.         }
  323.  
  324.         id = (id + N - 1) / 2;
  325.  
  326.         while (id > 0){
  327.             id--;
  328.             id /= 2;
  329.             pii p = fr1[id * 2 + 1];
  330.             if (fr1[id * 2 + 2].first > fr1[id * 2 + 1].first){
  331.                 p = fr1[id * 2 + 2];
  332.             }
  333.             fr1[id] = p;
  334.         }
  335.     }
  336.  
  337.     void upd(int id){
  338.         updMN(id);
  339.         updfr0(id);
  340.         updfr1(id);
  341.     }
  342. };
  343.  
  344. void slv(){
  345.     tree T;
  346.     T.bld();
  347.     if (sh){
  348.         cout<<"a\n"; cv(a);
  349.         T.showMN();
  350.         T.showfr0();
  351.         T.showfr1();
  352.     }
  353.     if (sh){
  354.         int q; cin>>q;
  355.         for (int i = 0; i < q; ++i){
  356.             int id; cin>>id;
  357.             T.upd(id);
  358.             if (sh){
  359.                 cout<<"a\n"; cv(a);
  360.                 T.showMN();
  361.                 T.showfr0();
  362.                 T.showfr1();
  363.             }
  364.  
  365.             //T.showfr1();
  366.         }
  367.     }
  368. }
  369.  
  370.  
  371. int main()
  372. {
  373.     /*ios::sync_with_stdio(0);
  374.     cin.tie(0);
  375.     cout.tie(0);*/
  376.     int t=1; if (!sh) cin>>t;
  377.     for (int go = 0; go < t; ++go){
  378.         cin>>n>>k; a.resize(n); for (int &i: a) cin>>i;
  379.         slv();
  380.     }
  381. }
  382. /*
  383. 20 20
  384. 1 4 3 6 3
  385. 1 4 7 4 7
  386. 1 4 3 6 4
  387. 1 4 3 2 3
  388. */
  389.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement