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

扫描二维码关注公众号，回复： 2370913 查看本文章

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 = 100000 + 100;
int arr[maxn];
int lazy[maxn << 2];
ll sum[maxn << 2];

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

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

void build(int rt, int l, int r)
{
    lazy[rt] = -1;          //注意，由于题目中Newp可能为0，所以这里不能默认为0,否则pushdown的时候，就会跳过NewP为0的操作  
    if (l==r)
    {
        sum[rt] = arr[l];
        return;
    }
    int mid = (l + r) / 2;
    build(rt << 1, l, mid);
    build(rt << 1 | 1, mid + 1, 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 + 1);
        return;
    }
    pushdown(rt, r - l + 1);
    int mid = (l + r) >> 1;
    if (L <= mid)update(L, R, C, l, mid, rt << 1);
    if (R > mid)update(L, R, C, mid + 1, r, rt << 1 | 1);
    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 + 1);
    int mid = (l + r) >> 1;
    ll ans = 0;
    if (L <= mid)ans += query(L, R, l, mid, rt << 1);
    if (R > mid)ans += query(L, R, mid + 1, r, rt << 1 | 1);
    return ans;
}


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

2018-07-25