Advertisement
tiom4eg

SIS Segtree

May 29th, 2022
104
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.31 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. // tiom4eg's precompiler options
  3. // POGGERS POGGERS POGGERS POGGERS POGGERS POGGERS POGGERS
  4. // Flags
  5. //#define MAXMIN
  6. #define ENDL
  7. #define AVX2
  8. #define PBDS
  9. // IO settings
  10. #define fastIO ios_base::sync_with_stdio(false); cin.tie(0)
  11. // Quick types
  12. #define ll long long
  13. #define ld double
  14. #define ull unsigned long long
  15. #define pii pair <int, int>
  16. #define pll pair <ll, ll>
  17. #define vi vector <int>
  18. #define mi vector <vector <int> >
  19. // Quick functions
  20. #ifdef ENDL
  21. #define endl "\n"
  22. #endif
  23. #define F first
  24. #define S second
  25. #define all(a) a.begin(), a.end()
  26. #define sz(a) a.size()
  27. #define pb push_back
  28. #define mp make_pair
  29. #ifdef MAXMIN
  30. #define min(a, b) ((a < b) ? (a) : (b))
  31. #define max(a, b) ((a > b) ? (a) : (b))
  32. #endif
  33. // Quick fors
  34. #define FOR(i, a, b) for (int i = a; i < b; ++i)
  35. #define RFOR(i, a, b) for (int i = a; i >= b; --i)
  36. // Pragmas
  37. #pragma GCC optimize("O3,unroll-loops") // let the chaos begin!
  38. #ifdef AVX2
  39. #pragma GCC target("avx,avx2,bmi,bmi2,popcnt,lzcnt,tune=native")
  40. #endif
  41. #pragma GCC comment(linker, "/stack:200000000")
  42. // PBDS
  43. #ifdef PBDS
  44. #include <ext/pb_ds/assoc_container.hpp>
  45. #include <ext/pb_ds/tree_policy.hpp>
  46. #define ordered_set tree <int, null_type, less <int>, rb_tree_tag, tree_order_statistics_node_update>
  47. #define ook order_of_key
  48. #define fbo find_by_order
  49. using namespace __gnu_pbds;
  50. #endif
  51. // POGGERS POGGERS POGGERS POGGERS POGGERS POGGERS POGGERS
  52. using namespace std;
  53. mt19937 rng(std::chrono::duration_cast<std::chrono::milliseconds>(std::chrono::system_clock::now().time_since_epoch()).count());
  54. #define int long long
  55. const int INF = 1e9 + 7, R = 1 << 17;
  56.  
  57. struct segtree {
  58.     struct node { int s, d; };
  59.     vector <node> t;
  60.     void merge(int v) { t[v].s = t[2 * v].s + t[2 * v + 1].s; }
  61.     void tag(int v, int x, int l) { t[v].s = l * x, t[v].d = x; }
  62.     void push(int v, int l) {
  63.         if (t[v].d == -INF) return;
  64.         tag(2 * v, t[v].d, l / 2), tag(2 * v + 1, t[v].d, l / 2);
  65.         t[v].d = -INF;
  66.     }
  67.     void build(vi& a) {
  68.         t.assign(2 * R, {0, -INF});
  69.         FOR(i, 0, sz(a)) t[R + i].s = a[i];
  70.         RFOR(i, R - 1, 1) merge(i);
  71.     }
  72.     void upd(int ql, int qr, int x, int v = 1, int nl = 0, int nr = R) { // set query
  73.         if (qr <= nl || nr <= ql) return;
  74.         if (ql <= nl && nr <= qr) {
  75.             tag(v, x, nr - nl);
  76.             return;
  77.         }
  78.         push(v, nr - nl);
  79.         int nm = (nl + nr) / 2;
  80.         upd(ql, qr, x, v * 2, nl, nm), upd(ql, qr, x, v * 2 + 1, nm, nr);
  81.         merge(v);
  82.     }
  83.     int get(int ql, int qr, int v = 1, int nl = 0, int nr = R) { // sum query
  84.         if (qr <= nl || nr <= ql) return 0;
  85.         if (ql <= nl && nr <= qr) return t[v].s;
  86.         push(v, nr - nl);
  87.         int nm = (nl + nr) / 2;
  88.         return get(ql, qr, v * 2, nl, nm) + get(ql, qr, v * 2 + 1, nm, nr);
  89.     }
  90. } t;
  91.  
  92. signed main() {
  93.     fastIO;
  94.     int n, m; cin >> n >> m;
  95.     vi a(n); FOR(i, 0, n) cin >> a[i];
  96.     int sum = 0; FOR(i, 0, n) sum += a[i];
  97.     t.build(a);
  98.     FOR(i, 0, m) {
  99.         int l, r; cin >> l >> r, --l;
  100.         int rsum = t.get(l, r);
  101.         ld g = (ld)(rsum) / (r - l);
  102.         int csum = t.get(0, R);
  103.         if (csum <= sum) t.upd(l, r, ceil(g));
  104.         else t.upd(l, r, floor(g));
  105.     }
  106.     FOR(i, 0, n) cout << t.get(i, i + 1) << ' ';
  107. }
  108.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement