Advertisement
cosenza987

Untitled

Feb 6th, 2023
1,123
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.74 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 = N) {
  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 = N) {
  26.         if(pcur == -1) {
  27.             pcur = roots[x] = cnt++;
  28.         }
  29.         if(p == -1) {
  30.             p = roots[x - 1];
  31.         }
  32.         seg[pcur].first = seg[p].first + 1;
  33.         seg[pcur].second = seg[p].second + val;
  34.         if(l == r) {
  35.             return;
  36.         }
  37.         int mid = (l + r) >> 1;
  38.         if(val > mid) {
  39.             L[pcur] = L[p];
  40.             R[pcur] = cnt++;
  41.             insert(x, val, R[pcur], R[p], mid + 1, r);
  42.         } else {
  43.             R[pcur] = R[p];
  44.             L[pcur] = cnt++;
  45.             insert(x, val, L[pcur], L[p], l, mid);
  46.         }
  47.     }
  48.     pair<int, long long> find(int p1, int p2, int i, int j, int l = 1, int r = N) {
  49.         if(i > r or j < l) {
  50.             return {0, 0};
  51.         }
  52.         if(i <= l and r <= j) {
  53.             return {seg[p1].first - seg[p2].first, seg[p1].second - seg[p2].second};
  54.         }
  55.         int mid = (l + r) >> 1;
  56.         auto a = find(L[p1], L[p2], i, j, l, mid);
  57.         auto b = find(R[p1], R[p2], i, j, mid + 1, r);
  58.         return {a.first + b.first, a.second + b.second};
  59.     }
  60. };
  61.  
  62. int query(int l, int r, int a, int b, long long s) {
  63.     int ans = 0, al = a, ar = b, lef = a, mn = -1;
  64.     long long dd = 0;
  65.     while(al <= ar) {
  66.         int mid = (al + ar) >> 1;
  67.         auto diff = pseg::find(pseg::roots[r], pseg::roots[l - 1], mid, b);
  68.         if(s * diff.first <= diff.second) {
  69.             ans = diff.first;
  70.             dd = diff.second;
  71.             lef = mid;
  72.             ar = mid - 1;
  73.         } else {
  74.             mn = mid;
  75.             al = mid + 1;
  76.         }
  77.     }
  78.     if(lef == a) {
  79.         return ans;
  80.     }
  81.     int add = (ans * s - dd) / (mn - s);
  82.     return ans + add;
  83. }
  84.  
  85. int main() {
  86.     ios_base::sync_with_stdio(false);
  87.     cin.tie(0);
  88.     int n;
  89.     cin >> n;
  90.     pseg::build();
  91.     pseg::roots[0] = 1;
  92.     for(int i = 1; i <= n; i++) {
  93.         int x;
  94.         cin >> x;
  95.         pseg::insert(i, x);
  96.     }
  97.     int q;
  98.     cin >> q;
  99.     while(q--) {
  100.         int l, r, a, b, s;
  101.         cin >> l >> r >> a >> b >> s;
  102.         cout << query(l, r, a, b, s) << "\n";
  103.     }
  104.     return 0;
  105. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement