Advertisement
IanisBelu

Y. Sneetches and Speeches 1

Dec 11th, 2023
46
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 5.01 KB | Source Code | 0 0
  1. #ifdef LOCAL
  2. #include <iostream>
  3. #include <fstream>
  4. #include <iomanip>
  5. #include <random>
  6. #include <vector>
  7. #include <queue>
  8. #include <stack>
  9. #include <set>
  10. #include <map>
  11. #else
  12. #pragma GCC optimize("Ofast,unroll-loops")
  13. #include <bits/stdc++.h>
  14. #define cerr if (false) cerr
  15. #define endl '\n'
  16. #endif
  17.  
  18. #define fi first
  19. #define se second
  20.  
  21. #define sz(a) ((int)(a).size())
  22. #define all(a) (a).begin(), (a).end()
  23.  
  24. #define lsb(x) (x & (-x))
  25.  
  26. using namespace std;
  27.  
  28. template <typename T>
  29. bool ckmax(T &a, T b) { return a < b ? a = b, true : false; }
  30. template <typename T>
  31. bool ckmin(T &a, T b) { return a > b ? a = b, true : false; }
  32.  
  33. using ll = long long;
  34. using pii = pair<int, int>;
  35.  
  36. const int NMAX = 3e5+5;
  37.  
  38. struct Treap {
  39.    bool val, lazy;
  40.    int max;
  41.    bool vall, valr;
  42.    int cntl, cntr;
  43.    int pr, size;
  44.    Treap *l, *r;
  45.    Treap(bool val): val(val), lazy(false), max(1), vall(val), valr(val), cntl(1), cntr(1), size(1), l(nullptr), r(nullptr) {
  46.       static mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
  47.       pr = int(rng());
  48.    }
  49. };
  50.  
  51. int tsize(Treap *a) { return a ? a->size : 0; }
  52. int tmax(Treap *a)  { return a ? a->max  : 0; }
  53.  
  54. Treap *tpropag(Treap *a) {
  55.    if (!a || !a->lazy) return a;
  56.  
  57.    a->val  ^= true;
  58.    a->vall ^= true;
  59.    a->valr ^= true;
  60.  
  61.    if (a->l) a->l->lazy ^= true;
  62.    if (a->r) a->r->lazy ^= true;
  63.  
  64.    a->lazy = false;
  65.    
  66.    return a;
  67. }
  68.  
  69. Treap *tupdate(Treap *a) {
  70.    if (!a) return a;
  71.  
  72.    a->size = tsize(a->l) + tsize(a->r) + 1;
  73.  
  74.    tpropag(a);
  75.    tpropag(a->l);
  76.    tpropag(a->r);
  77.  
  78.    if (a->l && a->r) {
  79.       bool lfull = tsize(a->l) == a->l->cntl;
  80.       bool rfull = tsize(a->r) == a->r->cntr;
  81.  
  82.       if (lfull && rfull && a->val == a->l->val && a->val == a->r->val) {
  83.          a->cntl = a->cntr = tsize(a);
  84.          a->vall = a->valr = a->val;
  85.          a->max = tsize(a);
  86.       } else {
  87.          a->cntl = a->l->cntl + (a->val == a->l->vall && lfull);
  88.          a->cntr = a->r->cntr + (a->val == a->r->valr && rfull);
  89.  
  90.          if (a->val == a->l->valr && a->val == a->r->vall) {
  91.             if (lfull)
  92.                a->cntl += a->r->cntl;
  93.             else if (rfull)
  94.                a->cntr += a->l->cntr;
  95.          }
  96.  
  97.          a->vall = a->l->vall;
  98.          a->valr = a->r->valr;
  99.  
  100.          a->max = a->l->cntr * (a->l->valr == a->val) + 1 + a->r->cntl * (a->r->vall == a->val);
  101.       }
  102.    } else if (a->l) {
  103.       a->cntl = a->l->cntl + (a->val == a->l->vall && a->l->cntl == tsize(a->l));
  104.       a->cntr = a->l->cntr * (a->val == a->l->valr) + 1;
  105.  
  106.       a->vall = a->l->vall;
  107.       a->valr = a->val;
  108.  
  109.       a->max = a->cntr;
  110.    } else if (a->r) {
  111.       a->cntl = a->r->cntl * (a->val == a->r->vall) + 1;
  112.       a->cntr = a->r->cntr + (a->val == a->r->valr && a->r->cntr == tsize(a->r));
  113.  
  114.       a->vall = a->val;
  115.       a->valr = a->r->valr;
  116.  
  117.       a->max = a->cntl;
  118.    } else {
  119.       a->cntl = a->cntr = 1;
  120.       a->vall = a->valr = a->val;
  121.       a->max = 1;
  122.    }
  123.  
  124.    ckmax(a->max, tmax(a->l));
  125.    ckmax(a->max, tmax(a->r));
  126.  
  127.    return a;
  128. }
  129.  
  130. ostream &operator<<(ostream &out, Treap *treap) {
  131.    function<void(Treap*)> dfs = [&](Treap *treap) {
  132.       if (!treap) return;
  133.       treap = tpropag(treap);
  134.       dfs(treap->l);
  135.       out << treap->val;
  136.       dfs(treap->r);
  137.    };
  138.    dfs(treap);
  139.    return out;
  140. }
  141.  
  142. Treap *join(Treap *a, Treap *b) {
  143.    if (!a || !b) return a ? a : b;
  144.    tpropag(a);
  145.    tpropag(b);
  146.    if (a->pr < b->pr) return a->r = join(a->r, b), tupdate(a);
  147.    return b->l = join(a, b->l), tupdate(b);
  148. }
  149.  
  150. Treap *join(vector<Treap*> v) {
  151.    if (v.empty()) return nullptr;
  152.    Treap *ret = v[0];
  153.    for (int i = 1; i < sz(v); i++)
  154.       ret = join(ret, v[i]);
  155.    return ret;
  156. }
  157.  
  158. pair<Treap*, Treap*> split(Treap *a, int size) {
  159.    if (!a) return { nullptr, nullptr };
  160.    tpropag(a);
  161.    if (tsize(a->l) + 1 <= size) {
  162.       auto [ l, r ] = split(a->r, size - tsize(a->l) - 1);
  163.       a->r = l;
  164.       return { tupdate(a), r };
  165.    }
  166.    auto [ l, r ] = split(a->l, size);
  167.    a->l = r;
  168.    return { l, tupdate(a) };
  169. }
  170.  
  171. int n, q;
  172. char a[NMAX];
  173.  
  174. void read() {
  175.    cin >> n >> q >> (a + 1);
  176. }
  177.  
  178. tuple<Treap*, Treap*, Treap*> split_segment(Treap *treap, int l, int r) {
  179.    auto [ a, suff ] = split(treap, l - 1);
  180.    auto [ mid, b  ] = split(suff, r - l + 1);
  181.    return { a, mid, b };
  182. }
  183.  
  184. Treap *update1(Treap *treap, int l, int r) {
  185.    auto [ a, mid, b ] = split_segment(treap, l, r);
  186.  
  187.    mid->lazy ^= true;
  188.  
  189.    return join({ a, mid, b });
  190. }
  191.  
  192. void solve() {
  193.    Treap *treap = new Treap(a[1] - '0');
  194.    for (int i = 2; i <= n; i++) {
  195.       treap = join(treap, new Treap(a[i] - '0'));
  196.    }
  197.  
  198.    while (q--) {
  199.       int t, l, r;
  200.       cin >> t >> l >> r;
  201.       if (t == 1) {
  202.          treap = update1(treap, l, r);
  203.       }
  204.       tupdate(treap);
  205.       cout << tmax(treap) << endl;
  206.    }
  207. }
  208.  
  209. signed main() {
  210. #ifdef LOCAL
  211.    freopen("input.txt", "r", stdin);
  212. #endif
  213.    ios_base::sync_with_stdio(false);
  214.    cin.tie(0);
  215.    cout.tie(0);
  216.  
  217.    read();
  218.    solve();
  219.  
  220.    return 0;
  221. }
  222.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement