HDU1698 线段树 区间更新

题意:区间更新及求和。
思路:模板题。

#include 
#include 
using namespace std;
const int MAXN = 100000 + 10;
int sum[MAXN << 2], add[MAXN << 2];
void PushUp(int rt)
{
    sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
    return;
}
void Build(int l, int r, int rt)
{
    if(l == r)
    {
        sum[rt] = 1;
        return;
    }
    int m = (l + r) >> 1;
    Build(l, m, rt << 1);
    Build(m + 1, r, rt << 1 | 1);
    PushUp(rt);
    return;
}
void PushDown(int ln, int rn, int rt)
{
    if(add[rt])
    {
        add[rt << 1] = add[rt];
        add[rt << 1 | 1] = add[rt];
        sum[rt << 1] = add[rt] * ln;
        sum[rt << 1 | 1] = add[rt] * rn;
        add[rt] = 0;
    }
}
void Update(int L, int R, int c, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        sum[rt] = c * (r - l + 1);
        add[rt] = c;
        return;
    }
    int m = (l + r) >> 1;
    PushDown(m - l + 1, r - m, rt);
    if(L <= m) Update(L, R, c, l, m, rt << 1);
    if(R > m) Update(L, R, c, m + 1, r, rt << 1 | 1);
    PushUp(rt);
    return;
}
int Query(int L, int R, int l, int r, int rt)
{
    if(L <= l && r <= R)
    {
        return sum[rt];
    }
    int m = (l + r) >> 1;
    PushDown(m - l + 1, r - m, rt);
    int ans = 0;
    if(L <= m) ans += Query(L, R, l, m, rt << 1);
    if(R > m) ans += Query(L, R, m + 1, r, rt << 1 | 1);
    return ans;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    int t; cin >> t;
    int kase = 0;
    while(t--)
    {
        int n; cin >> n;
        Build(1, n, 1);
        memset(add, 0, sizeof(add));
        int q; cin >> q;
        int x, y, z;
        for(int i = 0; i < q; i++)
        {
            cin >> x >> y >> z;
            Update(x, y, z, 1, n, 1);
        }
        cout << "Case " << ++kase << ": The total value of the hook is " << Query(1, n, 1, n, 1) << "." << endl;
    }
}

你可能感兴趣的:(----数据结构----,线段树)