hihoCoder week20 线段树的区间修改

区间修改 区间查询

最后一场比赛前的无可救药的热身

#include <bits/stdc++.h>
using namespace std;
#define mid ((l+r)/2)
#define ls (rt<<1)
#define rs (rt<<1|1)
const int N = 2e5+;

int n, s[N], tr[N], lz[N];

void build(int rt,int l,int r)
{
    if(l ==r) {
        tr[rt] = s[l];
        return ;
    }
    build(ls,l,mid);
    build(rs,mid+,r);
    tr[rt] = tr[ls] + tr[rs];
}

void down(int rt,int l,int r)
{
    if(lz[rt]) {
        lz[ls] = lz[rs] = lz[rt];
        tr[ls] = (mid-l+) * lz[ls];
        tr[rs] = (r - mid) * lz[rs];
        lz[rt] = ;
    }
} 

void update(int rt,int l,int r,int L,int R,int val)
{
    if(L <= l && r <= R) {
        lz[rt] = val;
        tr[rt] = val * (r-l+);
        return ;
    }
    down(rt,l,r);
    if(L<=mid)
        update(ls,l,mid,L,R,val);
    if(R>mid)
        update(rs,mid+,r,L,R,val);
    tr[rt] = tr[ls] + tr[rs];
}

long long query(int rt,int l,int r,int L,int R)
{
    if(L <= l && r <= R) {
        return tr[rt];
    }
    down(rt,l,r);
    long long res = ;
    if(L<=mid)
        res = res + query(ls,l,mid,L,R);
    if(R>mid)
        res = res + query(rs,mid+,r,L,R);
    return res;
}
void out(int rt,int l,int r)
{
    printf("%d %d %d\n", l, r, tr[rt]);
    if(l == r)
    return ;
    out(ls,l,mid);
    out(rs,mid+,r);
}
int main()
{
    freopen("in.txt", "r", stdin);
    scanf("%d", &n);
    for(int i=; i<=n; i++)
        scanf("%d", &s[i]);
    build(,,n);
    int m; scanf("%d", &m);
    while(m--) {
        int op,l,r;
        scanf("%d %d %d", &op, &l, &r);
        if(op) {
            int val; scanf("%d", &val);
            update(,,n,l,r,val);
            // out(1,1,n);
            //puts("-----");
        }else
            printf("%lld\n", query(,,n,l,r));
    }
    return ;
}