线段树模板

  1 #include<cstdio>
  2 #include<cstring>
  3 const int M = 30001;
  4 long long  sum[M << 2], a[M << 2];
  5 bool set[M << 2], flag;
  6 
  7 void init(void)
  8 {
  9     flag = 0;
 10     memset(sum, 0, sizeof(sum));
 11     memset(set, 0, sizeof(set));
 12     memset(a, 0, sizeof(a));
 13     
 14 }
 15 void add(int t, int x, int y)
 16 {
 17     if (set[t])
 18     {
 19         int m = (x + y) >> 1;
 20         set[t << 1] = set[t << 1 | 1] = set[t];
 21         a[t << 1] = a[t << 1 | 1] = a[t];
 22         sum[t << 1] = (m - x + 1) * a[t];
 23         sum[t << 1 | 1] = (y - m) * a[t];
 24         set[t] = 0;
 25     }
 26 }
 27 long long  query(int p, int q, int x, int y, int t)
 28 {
 29     if (p <= x&&y <= q)
 30         return sum[t];
 31     add(t, x, y);
 32     int m = (x + y) >> 1;
 33     long long  res = 0;
 34     if (p <= m) res += query(p, q, x, m, t << 1);
 35     if (q > m) res += query(p, q, m + 1, y, t << 1 | 1);
 36     return res;
 37 }
 38 void update(long long  op, int p, int q, int x, int y, int t)
 39 {
 40     if (p <= x&&y <= q)
 41     {
 42         set[t] = 1;
 43         a[t] = op;
 44         sum[t] = (y - x + 1)*op;
 45         return;
 46     }
 47     add(t, x, y);
 48     int m = (x + y) >> 1;
 49     if (p <= m) update(op, p, q, x, m, t << 1);
 50     if (q > m) update(op, p, q, m + 1, y, t << 1 | 1);
 51     sum[t] = sum[t << 1] + sum[t << 1 | 1];
 52 }
 53 void work(int x, int y, int t)
 54 {
 55     if (x == y)
 56     {
 57 
 58         if (flag)printf(" ");
 59         else flag = 1;
 60         printf("%lld", sum[t]);
 61         return;
 62     }
 63     add(t, x, y);
 64     int m = (x + y) >> 1;
 65     work(x, m, t << 1);
 66     work(m + 1, y, t << 1 | 1);
 67 }
 68 long long  calc(long long  t, int x, int y, bool up)
 69 {
 70     if (t >= 0)
 71     {
 72         if (up)return (t + y - x) / (y - x + 1);
 73         return t / (y - x + 1);
 74     }
 75     else
 76     {
 77         t = -t;
 78         if (!up)return -(t + y - x) / (y - x + 1);
 79         return -t / (y - x + 1);
 80     }
 81 }
 82 int main()
 83 {
 84     int n, m, x, y;
 85     long long  t, st, ori;
 86     while (~scanf("%d%d", &n, &m))
 87     {
 88         init();
 89         for (int i = 0; i < n; i++)
 90         {
 91             scanf("%lld", &t);
 92             update(t, i, i, 0, n - 1, 1);
 93         }
 94         ori = sum[1];
 95         while (m--)
 96         {
 97             scanf("%d%d", &x, &y);
 98             x--; y--;
 99             st = query(x, y, 0, n - 1, 1);
100             if (ori >= sum[1])update(calc(st, x, y, 1), x, y, 0, n - 1, 1);
101             else update(calc(st, x, y, 0), x, y, 0, n - 1, 1);
102         }
103         work(0, n - 1, 1);
104         printf("\n\n");
105     }
106     return 0;
107 }