Advertisement
cosenza987

Untitled

Feb 6th, 2023
987
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.99 KB | None | 0 0
  1. //Слава Україні, Героям слава
  2.  
  3. #include <bits/stdc++.h>
  4.  
  5. using namespace std;
  6.  
  7. const int N = 2e5 + 7, LOG = 18;
  8. const int NN = N * LOG;
  9.  
  10. namespace pseg {
  11.     pair<int, long long> seg[NN];
  12.     int roots[N], cnt = 1;
  13.     int L[NN], R[NN];
  14.     void build(int p = 1, int l = 1, int r = (1 << LOG)) {
  15.         if(l == r) {
  16.             seg[p] = {0, 0};
  17.             return;
  18.         }
  19.         int mid = (l + r) >> 1;
  20.         L[p] = cnt++;
  21.         R[p] = cnt++;
  22.         build(L[p], l, mid);
  23.         build(R[p], mid + 1, r);
  24.     }
  25.     void insert(int x, int val, int pcur = -1, int p = -1, int l = 1, int r = (1 << LOG)) {
  26.         if(pcur == -1) {
  27.             pcur = roots[x] = cnt++;
  28.         }
  29.         if(p == -1) {
  30.             p = roots[x - 1];
  31.         }
  32.         seg[pcur] = seg[p];
  33.         seg[pcur].first++; 
  34.         seg[pcur].second += val;
  35.         if(l == r) {
  36.             return;
  37.         }
  38.         int mid = (l + r) >> 1;
  39.         if(val > mid) {
  40.             L[pcur] = L[p];
  41.             R[pcur] = cnt++;
  42.             seg[R[pcur]] = seg[R[p]];
  43.             insert(x, val, R[pcur], R[p], mid + 1, r);
  44.         } else {
  45.             R[pcur] = R[p];
  46.             L[pcur] = cnt++;
  47.             seg[L[pcur]] = seg[L[p]];
  48.             insert(x, val, L[pcur], L[p], l, mid);
  49.         }
  50.     }
  51.     pair<int, long long> find(int p, int i, int j, int l = 1, int r = (1 << LOG)) {
  52.         if(i > r or j < l) {
  53.             return {0, 0};
  54.         }
  55.         if(i <= l and r <= j) {
  56.             return seg[p];
  57.         }
  58.         int mid = (l + r) >> 1;
  59.         auto a = find(L[p], i, j, l, mid);
  60.         auto b = find(R[p], i, j, mid + 1, r);
  61.         return {a.first + b.first, a.second + b.second};
  62.     }
  63. };
  64.  
  65. int query(int l, int r, int a, int b, long long s) {
  66.     int ans = 0, al = a, ar = b, lef = a, mn = -1;
  67.     long long dd = 0;
  68.     while(al <= ar) {
  69.         int mid = (al + ar) >> 1;
  70.         auto ll = pseg::find(pseg::roots[l - 1], mid, b);
  71.         auto rr = pseg::find(pseg::roots[r], mid, b);
  72.         if(s * (rr.first - ll.first) <= rr.second - ll.second) {
  73.             ans = rr.first - ll.first;
  74.             dd = rr.second - ll.second;
  75.             lef = mid;
  76.             ar = mid - 1;
  77.         } else {
  78.             mn = mid;
  79.             al = mid + 1;
  80.         }
  81.     }
  82.     if(lef == a) {
  83.         return ans;
  84.     }
  85.     auto ll = pseg::find(pseg::roots[l - 1], mn, b);
  86.     auto rr = pseg::find(pseg::roots[r], mn, b);
  87.     if(rr.first - ll.first == ans) {
  88.         return ans;
  89.     }
  90.     int add = (ans * s - dd) / (mn - s);
  91.     add = abs(add);
  92.     return ans + add;
  93. }
  94.  
  95. int main() {
  96.     ios_base::sync_with_stdio(false);
  97.     cin.tie(0);
  98.     int n;
  99.     cin >> n;
  100.     pseg::build();
  101.     for(int i = 1; i <= n; i++) {
  102.         int x;
  103.         cin >> x;
  104.         pseg::insert(i, x);
  105.     }
  106.     int q;
  107.     cin >> q;
  108.     while(q--) {
  109.         int l, r, a, b, s;
  110.         cin >> l >> r >> a >> b >> s;
  111.         cout << query(l, r, a, b, s) << "\n";
  112.     }
  113.     return 0;
  114. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement