hiho一下 第二十周（线段树模板）

基本上敲的模板

功能：区间更新，区间查询 

自己随意开发。。

#include "iostream"
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define bug  cout << "bug!!!" << endl;
const int MAXN = 100010;

int sum[MAXN<<2];
int p[MAXN<<2];

//更新父亲
void PushUp(int rt)
{
    sum[rt] = sum[rt<<1] + sum[rt<<1|1];
}

//下放
void PushDown(int rt,int m)
{
    if (p[rt]) {
        p[rt<<1] = p[rt<<1|1] = p[rt];
        sum[rt<<1] = (m-(m>>1)) * p[rt];
        sum[rt<<1|1] = (m >> 1) * p[rt];
        p[rt] = 0;
    }
}

void build(int l,int r, int rt)
{
    if (l == r) {
        cin >> p[rt];
        sum[rt] = p[rt];
        return;
    }
    int m = (l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}

//区间更新
void update(int L,int R,int c,int l,int r,int rt)
{
    if (L <= l && r <= R) {
        p[rt] = c;
        sum[rt] = c*(r-l+1);
        return;
    }
    PushDown(rt,r-l+1);
    int m = (l+r)>>1;
    if (L <= m) update(L,R,c,lson);
    if (R >  m) update(L,R,c,rson);
    PushUp(rt);
}

//区间查询
int query(int L, int R,int l,int r,int rt)
{
    if(L <= l && R >= r) {
        return sum[rt];
    }
    PushDown(rt,r-l+1);//比如rt存的1-4的和 而查询2-5 1-4就需要下放
    int m = (l + r) >> 1;
    int re = 0;
    if (L <= m) re += query(L,R,lson);
    if (R >  m) re += query(L,R,rson);
    return re;
}

int main()
{
    int n;
    cin >> n;
    build(1,n,1);
    int t;
    cin >> t;
    for (int i = 0;i < t;++ i) {
        int ok;
        cin >> ok;
        if (ok) {
            int p1,p2,c;
            cin >> p1 >> p2 >> c;
            update(p1,p2,c,1,n,1);
        }
        else {
            int p1,p2;
            cin >> p1 >> p2;
            cout << query(p1,p2,1,n,1) << endl;
        }
    }
}