hihocoder1078线段树修改区间查询区间（懒标记）

题目链接：http://hihocoder.com/problemset/problem/1078

我的代码：

  1 #include <iostream>
  2 #include <algorithm>
  3
  4 using namespace std;
  5
  6 #define MAXN 100005
  7
  8 int w[MAXN];
  9
 10 struct segNode
 11 {
 12     int left, right, sum;
 13     bool lazy;
 14 };
 15
 16 struct segTree
 17 {
 18     segNode t[4*MAXN];
 19     void build(int i, int l, int r)
 20     {
 21         t[i].left = l;
 22         t[i].right = r;
 23         t[i].lazy = false;
 24         if(l<r)
 25         {
 26             int m = (l+r)/2;
 27             build(2*i, l, m);
 28             build(2*i+1, m+1, r);
 29             t[i].sum = t[2*i].sum+t[2*i+1].sum;
 30         }
 31         else t[i].sum = w[l];
 32     }
 33     int query(int i, int l, int r)
 34     {
 35         if(t[i].lazy) pushdown(i);
 36         if(t[i].left==l&&t[i].right==r) return t[i].sum;
 37         int m = (t[i].left+t[i].right)/2;
 38         if(l>m) return query(2*i+1, l, r);
 39         if(r<=m) return query(2*i, l, r);
 40         return query(2*i, l, m)+query(2*i+1, m+1, r);
 41     }
 42     void update(int i, int l, int r, int v)
 43     {
 44         if(t[i].lazy) pushdown(i);
 45         if(t[i].left==l&&t[i].right==r)
 46         {
 47             t[i].sum = (t[i].right-t[i].left+1)*v;
 48             t[i].lazy = true;
 49         }
 50         else
 51         {
 52             int m = (t[i].left+t[i].right)/2;
 53             if(l>m) update(2*i+1, l, r, v);
 54             else if(r<=m) update(2*i, l, r, v);
 55             else
 56             {
 57                 update(2*i, l, m, v);
 58                 update(2*i+1, m+1, r, v);
 59             }
 60             t[i].sum = t[2*i].sum+t[2*i+1].sum;
 61         }
 62     }
 63     void pushdown(int i)
 64     {
 65         t[i].lazy = false;
 66         if(t[i].left<t[i].right)
 67         {
 68             int l = t[i].right-t[i].left+1;
 69             int ll = t[2*i].right-t[2*i].left+1;
 70             int rl = t[2*i+1].right-t[2*i+1].left+1;
 71             t[2*i].sum = t[i].sum/l*ll;
 72             t[2*i+1].sum = t[i].sum/l*rl;
 73             t[2*i].lazy = true;
 74             t[2*i+1].lazy = true;
 75         }
 76     }
 77 }tree;
 78
 79 int main()
 80 {
 81     int n, q;
 82     bool op;
 83     while(cin>>n)
 84     {
 85         for(int i=1; i<=n; ++i) cin>>w[i];
 86         tree.build(1, 1, n);
 87         cin>>q;
 88         while(q--)
 89         {
 90             cin>>op;
 91             if(op)
 92             {
 93                 int l, r, v;
 94                 cin>>l>>r>>v;
 95                 tree.update(1, l, r, v);
 96             }
 97             else
 98             {
 99                 int l, r;
100                 cin>>l>>r;
101                 cout<<tree.query(1, l, r)<<endl;
102             }
103         }
104     }
105     return 0;
106 }