HihoCoder - 1078 【区间修改】

题目链接：https://vjudge.net/contest/241135#problem/A

题目大意：（与区间修改模板题相同）

输入

每个测试点（输入文件）有且仅有一组测试数据。

每组测试数据的第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&ltPi, NewP<=10^4。

输出

对于每组测试数据，对于每个小Hi的询问，按照在输入中出现的顺序，各输出一行，表示查询的结果：标号在区间[Li, Ri]中的所有商品的价格之和。

Sample Input

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

Sample Output

4731
14596

此题虽然只是区间修改的模板题，但是要注意修改值可以为0，所以lazy的初始值不能置为0，否则会将初始值与修改值混淆。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
const int maxn =  + ;
int arr[maxn];
int lazy[maxn << ];
ll sum[maxn << ];

void PushUp(int rt)
{
    sum[rt] = sum[rt << ] + sum[rt <<  | ];
}

void pushdown(int rt, int len)
{
    if (lazy[rt]!=-)         //如果该节点的lazy发生改变
    {
        lazy[rt << ] = lazy[rt];
        lazy[rt <<  | ] = lazy[rt];
        sum[rt << ] = (ll)lazy[rt] * (len - (len >> ));
        sum[rt <<  | ] = (ll)lazy[rt] * (len >> );
        lazy[rt] = -;         //!!!!这里一定不能忘记
    }
}

void build(int rt, int l, int r)
{
    lazy[rt] = -;          //注意，由于题目中Newp可能为0，所以这里不能默认为0,否则pushdown的时候，就会跳过NewP为0的操作
    if (l==r)
    {
        sum[rt] = arr[l];
        return;
    }
    int mid = (l + r) / ;
    build(rt << , l, mid);
    build(rt <<  | , mid + , r);
    PushUp(rt);
}

void update(int L, int R, int C, int l, int r, int rt)
{
    if (L <= l && r <= R)
    {
        lazy[rt] = C;
        sum[rt] = (ll)C * (r - l + );
        return;
    }
    pushdown(rt, r - l + );
    int mid = (l + r) >> ;
    if (L <= mid)update(L, R, C, l, mid, rt << );
    if (R > mid)update(L, R, C, mid + , r, rt <<  | );
    PushUp(rt);
}

ll query(int L, int R, int l, int r, int rt)
{
    if (L <= l && r <= R)
    {
        return sum[rt];
    }
    pushdown(rt, r - l + );
    int mid = (l + r) >> ;
    ll ans = ;
    if (L <= mid)ans += query(L, R, l, mid, rt << );
    if (R > mid)ans += query(L, R, mid + , r, rt <<  | );
    return ans;
}

int main()
{
    int n;
    while (scanf("%d", &n) != EOF) {
        for (int i = ; i <= n; i++)
        {
            scanf("%d", &arr[i]);
        }
        build(, , n);
        int m; scanf("%d", &m);
        for (int i = ; i < m; i++)
        {
            int a; scanf("%d", &a);
            if (a == )
            {
                int x, y, z;
                scanf("%d%d%d", &x, &y, &z);
                update(x, y, z, , n, );
            }
            else
            {
                int x, y;
                scanf("%d%d", &x, &y);
                printf("%lld\n", query(x, y, , n, ));
            }
        }
    }
    return ;
}

2018-07-25