線段樹的區間改動

描寫敘述

對於小Ho表現出的對線段樹的理解。小Hi表示挺愜意的，可是愜意就夠了麽？於是小Hi將問題改了改。又出給了小Ho：

如果貨架上從左到右擺放了N種商品。而且依次標號為1到N，當中標號為i的商品的價格為Pi。小Hi的每次操作分為兩種可能。第一種是改動價格——小Hi給出一段區間[L, R]和一個新的價格NewP，全部標號在這段區間中的商品的價格都變成NewP。

另外一種操作是詢問——小Hi給出一段區間[L, R]。而小Ho要做的便是計算出全部標號在這段區間中的商品的總價格。然後告訴小Hi。

那麽這種一個問題，小Ho該怎樣解決呢？

提示：推動科學發展的除了人的好奇心之外還有人的懶惰心！

輸入

每一個測試點（輸入文件）有且僅有一組測試數據。

每組測試數據的第1行為一個整數N，意義如前文所述。

每組測試數據的第2行為N個整數，分別描寫敘述每種商品的重量。當中第i個整數表示標號為i的商品的重量Pi。

每組測試數據的第3行為一個整數Q，表示小Hi進行的操作數。

每組測試數據的第N+4~N+Q+3行，每行分別描寫敘述一次操作，每行的開頭均為一個屬於0或1的數字，分別表示該行描寫敘述一個詢問和一次商品的價格的更改兩種情況。

對於第N+i+3行。假設該行描寫敘述一個詢問。則接下來為兩個整數Li, Ri。表示小Hi詢問的一個區間[Li, Ri]；假設該行描寫敘述一次商品的價格的更改。則接下來為三個整數Li，Ri，NewP，表示標號在區間[Li, Ri]的商品的價格所有改動為NewP。

對於100%的數據。滿足N<=10^5，Q<=10^5, 1<=Li<=Ri<=N，1<=Pi<=N, 0<Pi, NewP<=10^4。

輸出

對於每組測試數據。對於每一個小Hi的詢問，依照在輸入中出現的順序。各輸出一行，表示查詢的結果：標號在區間[Li, Ri]中的全部商品的價格之和。

例子輸入

10
4733 6570 8363 7391 4511 1433 2281 187 5166 378 
6
1 5 10 1577
1 1 7 3649
0 8 10
0 1 4
1 6 8 157
1 3 4 1557

例子輸出

4731
14596

#include <stdio.h>
#include <string.h>

#define maxn 100002
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1

struct Node {
    int sum, lazy;
} T[maxn << 2];

void pushUp(int rt) {
    T[rt].sum = T[rt<<1].sum + T[rt<<1|1].sum;
}

void pushDown(int l, int r, int rt) {
    int mid = (l + r) >> 1;
    T[rt<<1].sum = T[rt].lazy * (mid - l + 1);
    T[rt<<1|1].sum = T[rt].lazy * (r - mid);
    T[rt<<1].lazy = T[rt].lazy;
    T[rt<<1|1].lazy = T[rt].lazy;
    T[rt].lazy = 0;
}

void build(int l, int r, int rt) {
    if(l == r) {
        scanf("%d", &T[rt].sum);
        return;
    }
    int mid = (l + r) >> 1;
    build(lson);
    build(rson);
    pushUp(rt);
}

void update(int L, int R, int V, int l, int r, int rt) {
    if(l == L && r == R) {
        T[rt].lazy = V;
        T[rt].sum = V * (r - l + 1);
        return;
    }
    int mid = (l + r) >> 1;
    if(T[rt].lazy) pushDown(l, r, rt);
    if(R <= mid) update(L, R, V, lson);
    else if(L > mid) update(L, R, V, rson);
    else {
        update(L, mid, V, lson);
        update(mid + 1, R, V, rson);
    }
    pushUp(rt);
}

int query(int L, int R, int l, int r, int rt) {
    if(l == L && R == r)
        return T[rt].sum;
    int mid = (l + r) >> 1;
    if(T[rt].lazy) pushDown(l, r, rt);
    if(R <= mid) return query(L, R, lson);
    else if(L > mid) return query(L, R, rson);
    return query(L, mid, lson) + query(mid + 1, R, rson);
}

int main() {
    int N, Q, i, a, b, c, d;
    scanf("%d", &N);
    build(1, N, 1);
    scanf("%d", &Q);
    while(Q--) {
        scanf("%d%d%d", &a, &b, &c);
        if(a) {
            scanf("%d", &d);
            update(b, c, d, 1, N, 1);
        } else printf("%d\n", query(b, c, 1, N, 1));
    }
    return 0;
}

線段樹的區間改動