HDU 1698 Just A Hook

最新推荐文章于 2022-12-12 21:12:45 发布

原创最新推荐文章于 2022-12-12 21:12:45 发布 · 328 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一道经典的线段树区间修改问题，通过具体实例展示了如何使用线段树进行区间更新并求区间和，提供了完整的C++实现代码。

题目链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=1698

大意：给出n个数，初始值都为1，操作（x，y，z）表示把区间[x,y]上的数都变为z。最后求整个区间和。

线段树区间修改值的模板题。。。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define lc rt<<1
#define rc rt<<1|1
using namespace std;

typedef long long ll;
const int maxn = 100000 + 10;
const int INF = 1e9 + 10;

int sum[maxn<<2],var[maxn<<2],n;

void PushUp(int rt)
{
    sum[rt] = sum[lc] + sum[rc];
}

void PushDown(int rt, int m)
{
    if (var[rt]) {
       var[lc] = var[rc] = var[rt];
       sum[lc] = var[lc] * (m - (m >> 1));
       sum[rc] = var[rc] * (m >> 1);
       var[rt] = 0; 
    }
    
}

void build(int l, int r, int rt)
{
    if (l == r) {
        sum[rt] = 1;
    }
    else {
        int m = (l + r) >> 1;
        build(lson);
        build(rson);
        PushUp(rt);
    }
}

void update(int ql, int qr, int c, int l, int r, int rt)
{
    if (ql <= l && r <= qr) {
        var[rt] = c;
        sum[rt] = c * (r-l+1);
        return;
    }
    PushDown(rt, r-l+1);
    int m = (l + r) >> 1;
    if (ql <= m) update(ql, qr, c, lson);
    if (m < qr) update(ql, qr, c, rson);
    PushUp(rt);
}

int main()
{
    int T, kase = 0;
    scanf("%d",&T);
    while(T--) {
        memset(sum, 0, sizeof sum);
        memset(var, 0, sizeof var);
        scanf("%d",&n);
        build(1,n,1);
        int q;
        scanf("%d",&q);
        while(q--) {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
        printf("Case %d: The total value of the hook is %d\n",++kase,sum[1]);
    }
    return 0;
}