leoanjos

Range Min With

Dec 14th, 2021
179
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.50 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2.  
  3. using namespace std;
  4.  
  5. #define long long long int
  6.  
  7. struct SegmentTree {
  8. private:
  9.     struct Node {
  10.         int sum, mx, snd_mx, cnt;
  11.  
  12.         Node(): sum(0), mx(0), snd_mx(0), cnt(0) {}
  13.         Node(int sum, int mx, int snd_mx, int cnt):
  14.             sum(sum), mx(mx), snd_mx(snd_mx), cnt(cnt) {}
  15.  
  16.         Node operator +(Node other) {
  17.             int new_sum = sum + other.sum;
  18.  
  19.             int new_mx = mx, new_cnt = cnt;
  20.             if (other.mx == mx) new_cnt += other.cnt;
  21.             else if (other.mx > mx) {
  22.                 new_mx = other.mx;
  23.                 new_cnt = other.cnt;
  24.             }
  25.  
  26.             int new_snd_mx = -1;
  27.             if (mx > other.mx) new_snd_mx = max(snd_mx, other.mx);
  28.             else if (mx < other.mx) new_snd_mx = max(mx, other.snd_mx);
  29.             else new_snd_mx = max(snd_mx, other.snd_mx);
  30.  
  31.             return Node(new_sum, new_mx, new_snd_mx, new_cnt);
  32.         }
  33.     };
  34.  
  35.     int n;
  36.     vector<int> a;
  37.     vector<Node> tree;
  38.     vector<int> lazy;
  39.  
  40. public:
  41.     SegmentTree(int n, vector<int> &a) {
  42.         this->n = n;
  43.         this->a = a;
  44.  
  45.         tree.resize(4 * n);
  46.         lazy.assign(4 * n, -1);
  47.         build(1, 1, n);
  48.     }
  49.  
  50.     void update(int l, int r, int x) {
  51.         update(1, 1, n, l, r, x);
  52.     }
  53.  
  54.     int query(int l, int r) {
  55.         return query(1, 1, n, l, r);
  56.     }
  57.  
  58. private:
  59.     void merge(int node) {
  60.         tree[node] = tree[2 * node] + tree[2 * node + 1];
  61.     }
  62.  
  63.     void build(int node, int l, int r) {
  64.         if (l == r) tree[node] = Node(a[l], a[l], -1, 1);
  65.         else {
  66.             int mid = (l + r) / 2;
  67.             build(2 * node, l, mid);
  68.             build(2 * node + 1, mid + 1, r);
  69.             merge(node);
  70.         }
  71.     }
  72.  
  73.     void update_lazy(int node, int l, int r, int x) {
  74.         if (x <= tree[node].mx) {
  75.             tree[node].sum -= tree[node].cnt * (tree[node].mx - x);
  76.             tree[node].mx = lazy[node] = x;
  77.         }
  78.     }
  79.  
  80.     void push_down(int node, int l, int r) {
  81.         if (lazy[node] == -1) return;
  82.  
  83.         int mid = (l + r) / 2;
  84.         update_lazy(2 * node, l, mid, lazy[node]);
  85.         update_lazy(2 * node + 1, mid + 1, r, lazy[node]);
  86.         lazy[node] = -1;
  87.     }
  88.  
  89.     void update(int node, int l, int r, int ul, int ur, int x) {
  90.         if (r < ul || l > ur || x > tree[node].mx) return;
  91.         if (ul <= l && r <= ur && x > tree[node].snd_mx) {
  92.             update_lazy(node, l, r, x);
  93.             return;
  94.         }
  95.  
  96.         push_down(node, l, r);
  97.  
  98.         int mid = (l + r) / 2;
  99.         update(2 * node, l, mid, ul, ur, x);
  100.         update(2 * node + 1, mid + 1, r, ul, ur, x);
  101.         merge(node);
  102.     }
  103.  
  104.     int query(int node, int l, int r, int ql, int qr) {
  105.         if (r < ql || l > qr) return 0;
  106.         if (ql <= l && r <= qr) return tree[node].sum;
  107.  
  108.         push_down(node, l, r);
  109.  
  110.         int mid = (l + r) / 2;
  111.         return query(2 * node, l, mid, ql, qr) + query(2 * node + 1, mid + 1, r, ql, qr);
  112.     }
  113. };
  114.  
  115. int main() {
  116.     ios_base::sync_with_stdio(false);
  117.     cin.tie(NULL);
  118.  
  119.     int n; cin >> n;
  120.  
  121.     vector<int> a(n + 1);
  122.     for (int i = 1; i <= n; i++)
  123.         cin >> a[i];
  124.  
  125.     SegmentTree tree(n, a);
  126.  
  127.     int q; cin >> q;
  128.     while (q--) {
  129.         int op, l, r;
  130.         cin >> op >> l >> r;
  131.  
  132.         if (op == 1) {
  133.             int ans = tree.query(l, r);
  134.             cout << ans << "\n";
  135.         } else {
  136.             int x; cin >> x;
  137.             tree.update(l, r, x);
  138.         }
  139.     }
  140. }
Advertisement
Add Comment
Please, Sign In to add comment