EntropyIncreaser

TJOI2016 sort

Nov 24th, 2017
207
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.83 KB | None | 0 0
  1. #include <cstdio>
  2.  
  3. #define DEBUG() // do {debug(root); LOG("\n"); } while (false)
  4. #define LOG(FMT...) // fprintf(stderr, FMT)
  5.  
  6. using namespace std;
  7.  
  8. struct node {
  9.     int l, r, d, lazy;
  10.     node *ls, *rs;
  11. };
  12.  
  13. const int N = 100010;
  14.  
  15. int n, m, q;
  16. int a[N];
  17. bool op[N];
  18. int opl[N], opr[N];
  19. node* root;
  20.  
  21. node* build(int l, int r);
  22. void part(node* p, int x);
  23. int query(node* p, int l, int r);
  24. void change(node* p, int l, int r, int x);
  25. void push_down(node* p);
  26. int get_d(node* p);
  27. void debug(node* p);
  28.  
  29. bool pred(int x);
  30.  
  31. int main() {
  32.     scanf("%d%d", &n, &m);
  33.     for (int i = 1; i <= n; ++i)
  34.         scanf("%d", &a[i]);
  35.     for (int i = 1; i <= m; ++i)
  36.         scanf("%d%d%d", &op[i], &opl[i], &opr[i]);
  37.     scanf("%d", &q);
  38.    
  39.     root = build(1, n);
  40.    
  41.     int l = 1, r = n, mid;
  42.     while (l < r) {
  43.         mid = (l + r) >> 1;
  44.         if (pred(mid))
  45.             l = mid + 1;
  46.         else
  47.             r = mid;
  48.     }
  49.     printf("%d\n", l);
  50.    
  51. //  while (getchar());
  52.    
  53.     return 0;
  54. }
  55.  
  56. node* build(int l, int r) {
  57.     static node pool[N << 1];
  58.     static node* pc = pool;
  59.     node* p = pc;
  60.     ++pc;
  61.     p->l = l;
  62.     p->r = r;
  63.     if (l == r)
  64.         return p;
  65.     int mid = (l + r) >> 1;
  66.     p->ls = build(l, mid);
  67.     p->rs = build(mid + 1, r);
  68.     return p;
  69. }
  70.  
  71. void part(node* p, int x) {
  72.     p->lazy = 2;
  73.     if (p->l == p->r) {
  74.         p->d = a[p->l] > x;
  75.         return;
  76.     }
  77.     part(p->ls, x);
  78.     part(p->rs, x);
  79.     p->d = p->ls->d + p->rs->d;
  80. }
  81.  
  82. inline bool pred(int x) {
  83.     part(root, x);
  84.     int cnt;
  85.     DEBUG();
  86.     for (int i = 1; i <= m; ++i) {
  87.         LOG("[INFO]: %d %d %d\n", op[i], opl[i], opr[i]);
  88.         cnt = query(root, opl[i], opr[i]);
  89.         if (op[i]) {
  90.             change(root, opl[i], opl[i] + cnt - 1, 1);
  91.             change(root, opl[i] + cnt, opr[i], 0);
  92.         } else {
  93.             change(root, opl[i], opr[i] - cnt, 0);
  94.             change(root, opr[i] - cnt + 1, opr[i], 1);
  95.         }
  96.         DEBUG();
  97.     }
  98.     return query(root, q, q);
  99. }
  100.  
  101. int query(node* p, int l, int r) {
  102.     if (p->l == l && p->r == r)
  103.         return get_d(p);
  104.     push_down(p);
  105.     if (r <= p->ls->r)
  106.         return query(p->ls, l, r);
  107.     if (l >= p->rs->l)
  108.         return query(p->rs, l, r);
  109.     return query(p->ls, l, p->ls->r) +
  110.            query(p->rs, p->rs->l, r);
  111. }
  112.  
  113. void change(node* p, int l, int r, int x) {
  114.     if (l > r)
  115.         return;
  116.     if (p->l == l && p->r == r) {
  117.         p->lazy = x;
  118.         return;
  119.     }
  120.     push_down(p);
  121.     if (r <= p->ls->r)
  122.         change(p->ls, l, r, x);
  123.     else if (l >= p->rs->l)
  124.         change(p->rs, l, r, x);
  125.     else {
  126.         change(p->ls, l, p->ls->r, x);
  127.         change(p->rs, p->rs->l, r, x);
  128.     }
  129.     p->d = get_d(p->ls) + get_d(p->rs);
  130. }
  131.  
  132. inline void push_down(node* p) {
  133.     if (p->lazy == 2)
  134.         return;
  135.     p->ls->lazy = p->rs->lazy = p->lazy;
  136.     p->d = p->lazy ? (p->r - p->l + 1) : 0;
  137.     p->lazy = 2;
  138. }
  139.  
  140. inline int get_d(node* p) {
  141.     if (p->lazy == 2)
  142.         return p->d;
  143.     return p->lazy ? (p->r - p->l + 1) : 0;
  144. }
  145.  
  146. void debug(node* p) {
  147.     if (p->l == p->r) {
  148.         LOG("%2d", get_d(p));
  149.         return;
  150.     }
  151.     push_down(p);
  152.     debug(p->ls);
  153.     debug(p->rs);
  154. }
Advertisement
Add Comment
Please, Sign In to add comment